Home
last modified time | relevance | path

Searched refs:ProcessSubcomputation (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_cost_analysis.cc373 ProcessSubcomputation(map->to_apply())); in HandleMap()
389 ProcessSubcomputation(function)); in HandleReduce()
415 ProcessSubcomputation(function)); in HandleReduceWindow()
443 ProcessSubcomputation(instruction->select())); in HandleSelectAndScatter()
445 ProcessSubcomputation(instruction->scatter())); in HandleSelectAndScatter()
490 ProcessSubcomputation(async_start->called_computations()[0])); in HandleAsyncStart()
883 ProcessSubcomputation(fusion->fused_instructions_computation())); in HandleFusion()
989 ProcessSubcomputation(call->to_apply())); in HandleCall()
1023 ProcessSubcomputation(xla_while->while_body())); in HandleWhile()
1026 ProcessSubcomputation(xla_while->while_condition())); in HandleWhile()
[all …]
H A Dhlo_cost_analysis.h252 StatusOr<Properties> ProcessSubcomputation(HloComputation* computation);