Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc266 ProcessNestedSubcomputation(map->to_apply())); in HandleMap()
282 ProcessNestedSubcomputation(function)); in HandleReduce()
308 ProcessNestedSubcomputation(function)); in HandleReduceWindow()
333 ProcessNestedSubcomputation(instruction->select())); in HandleSelectAndScatter()
335 ProcessNestedSubcomputation(instruction->scatter())); in HandleSelectAndScatter()
617 ProcessNestedSubcomputation(fusion->fused_instructions_computation())); in HandleFusion()
727 ProcessNestedSubcomputation(scatter->to_apply())); in HandleScatter()
778 HloCostAnalysis::ProcessNestedSubcomputation(HloComputation* computation) { in ProcessNestedSubcomputation() function in xla::HloCostAnalysis
Dhlo_cost_analysis.h176 StatusOr<Properties> ProcessNestedSubcomputation(HloComputation* computation);