Searched refs:map_defun_fn_ (Results 1 – 1 of 1) sorted by relevance
222 std::unique_ptr<FunctionBody> map_defun_fn_; member in tensorflow::grappler::vectorization_utils::__anon1f3551c50111::Vectorization268 TF_RETURN_IF_ERROR(AddMapDefunOutput(map_defun_fn_.get(), map_defun_node_, in AddConversionMapping()270 int output_index = map_defun_fn_->ret_nodes.size() - 1; in AddConversionMapping()302 map_defun_fn_->ret_nodes[output_position]->input_edge(0, &ret_edge)); in ConvertOutput()326 RemoveMapDefunOutput(output_position, outer_scope_.get(), map_defun_fn_.get(), in ConvertOutput()346 map_defun_fn_->ret_nodes); in VectorizeHelper()353 Node* output_node = map_defun_fn_->ret_nodes.at(output_position); in VectorizeHelper()362 if (map_defun_fn_->ret_nodes.empty()) { in VectorizeHelper()395 map_defun_fn_.reset(tmp); in Initialize()526 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], true)); in AddArgTensorMappings()[all …]