Searched refs:map_defun_node_ (Results 1 – 1 of 1) sorted by relevance
223 Node* map_defun_node_ = nullptr; // Owned by `outer_scope` member in tensorflow::grappler::vectorization_utils::__anoncb11dea30111::Vectorization268 TF_RETURN_IF_ERROR(AddMapDefunOutput(map_defun_fn_.get(), map_defun_node_, in AddConversionMapping()271 inputs.push_back({map_defun_node_, output_index, true}); in AddConversionMapping()324 ReplaceEdgeSources({map_defun_node_, output_position}, converted_output, in ConvertOutput()327 map_defun_node_); in ConvertOutput()363 outer_scope_->RemoveNode(map_defun_node_); in VectorizeHelper()398 map_defun_node_ = outer_scope_->FindNodeId(node_id); in Initialize()438 TF_RETURN_IF_ERROR(map_defun_node_->input_node(0, &input_node)); in StackTensor()501 map_defun_node_->attrs().Find("Targuments")->list().type_size(); in AddArgTensorMappings()505 TF_RETURN_IF_ERROR(map_defun_node_->input_node( in AddArgTensorMappings()[all …]