Searched refs:map_defun_fn_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 222 std::unique_ptr<FunctionBody> map_defun_fn_; member in tensorflow::grappler::vectorization_utils::__anoncb11dea30111::Vectorization
268 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()
388 FunctionDefToBodyHelper(*map_defun_fn, {}, &lib_def_, &map_defun_fn_)); in Initialize()
519 TF_RETURN_IF_ERROR(add_conversion(map_defun_fn_->arg_nodes[i], true)); in AddArgTensorMappings()
[all …]
|