Searched refs:embedded_evaluator (Results 1 – 2 of 2) sorted by relevance
1216 HloEvaluator embedded_evaluator; in HandleCall() local1217 embedded_evaluator.set_dynamic_dimension_inference( in HandleCall()1220 embedded_evaluator.Evaluate(*computation, arg_literals)); in HandleCall()1251 HloEvaluator embedded_evaluator; in HandleFusion() local1252 embedded_evaluator.set_dynamic_dimension_inference( in HandleFusion()1254 TF_ASSIGN_OR_RETURN(Literal result, embedded_evaluator.Evaluate( in HandleFusion()1276 HloEvaluator embedded_evaluator; in HandleConditional() local1277 embedded_evaluator.set_dynamic_dimension_inference( in HandleConditional()1280 embedded_evaluator.Evaluate( in HandleConditional()1463 HloEvaluator embedded_evaluator(max_loop_iterations_); in HandleSort() local[all …]
1589 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in MapImpl()1608 embedded_evaluator.Evaluate(*computation, arg_literals) in MapImpl()1612 embedded_evaluator.ResetVisitStates(); in MapImpl()1738 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in HandleReduce()1807 embedded_evaluator.Evaluate( in HandleReduce()1811 embedded_evaluator.ResetVisitStates(); in HandleReduce()1887 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_); in HandleSelectAndScatter()1922 embedded_evaluator in HandleSelectAndScatter()1931 embedded_evaluator.ResetVisitStates(); in HandleSelectAndScatter()1944 embedded_evaluator in HandleSelectAndScatter()[all …]