Searched refs:embedded_evaluator (Results 1 – 2 of 2) sorted by relevance
1928 HloEvaluator embedded_evaluator; in HandleCall() local1929 embedded_evaluator.set_dynamic_dimension_inference( in HandleCall()1932 embedded_evaluator.Evaluate(*computation, arg_literals)); in HandleCall()1963 HloEvaluator embedded_evaluator; in HandleFusion() local1964 embedded_evaluator.set_dynamic_dimension_inference( in HandleFusion()1966 TF_ASSIGN_OR_RETURN(Literal result, embedded_evaluator.Evaluate( in HandleFusion()1988 HloEvaluator embedded_evaluator; in HandleConditional() local1989 embedded_evaluator.set_dynamic_dimension_inference( in HandleConditional()1992 embedded_evaluator.Evaluate( in HandleConditional()2183 HloEvaluator embedded_evaluator(max_loop_iterations_); in HandleSort() local[all …]
1733 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);1752 embedded_evaluator.Evaluate(*computation, arg_literals)1756 embedded_evaluator.ResetVisitStates();1871 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);1906 embedded_evaluator1915 embedded_evaluator.ResetVisitStates();1928 embedded_evaluator1935 embedded_evaluator.ResetVisitStates();1985 HloEvaluator embedded_evaluator(parent_->max_loop_iterations_);2014 computed_result[0] = embedded_evaluator.Evaluate(*function, args)[all …]