Searched refs:map_defun_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 69 void UpdateMapDefunAttrs(FunctionBody* map_defun_fn, Node* map_defun_node) { in UpdateMapDefunAttrs() argument
70 map_defun_node->AddAttr("output_types", map_defun_fn->ret_types); in UpdateMapDefunAttrs()
74 map_defun_node->AddAttr("output_shapes", std::vector<PartialTensorShape>( in UpdateMapDefunAttrs()
78 Status AddMapDefunOutput(FunctionBody* map_defun_fn, Node* map_defun_node, in AddMapDefunOutput() argument
96 UpdateMapDefunAttrs(map_defun_fn, map_defun_node); in AddMapDefunOutput()
102 FunctionBody* map_defun_fn, Node* map_defun_node) { in RemoveMapDefunOutput() argument
115 UpdateMapDefunAttrs(map_defun_fn, map_defun_node); in RemoveMapDefunOutput()
119 ReplaceEdgeSources({map_defun_node, output_position + i + 1}, in RemoveMapDefunOutput()
120 {map_defun_node, output_position + i}, outer_scope); in RemoveMapDefunOutput()
144 const NodeDef& map_defun_node, FunctionDef** result);
[all …]
|
| D | map_vectorization.cc | 70 NodeDef* map_defun_node = vectorized_func->mutable_node_def()->Add(); in CreateMapDefunWrapper() local
71 map_defun_node->set_op("MapDefun"); in CreateMapDefunWrapper()
72 function_utils::SetUniqueFunctionNodeName(map_defun_node->op(), in CreateMapDefunWrapper()
73 vectorized_func, map_defun_node); in CreateMapDefunWrapper()
79 graph_utils::CopyAttribute(k, map_node, map_defun_node); in CreateMapDefunWrapper()
94 map_defun_node->add_input(input.name()); in CreateMapDefunWrapper()
98 AddNodeAttr("Targuments", t_args, map_defun_node); in CreateMapDefunWrapper()
99 AddNodeAttr("Tcaptured", t_captured, map_defun_node); in CreateMapDefunWrapper()
102 string output_prefix = strings::StrCat(map_defun_node->name(), ":output:"); in CreateMapDefunWrapper()
119 const NodeDef& map_defun_node = vectorized_func->node_def(0); in AddVectorizedFunction() local
[all …]
|
| D | vectorization_utils_test.cc | 70 Node* map_defun_node; in WrapFunctionWithMapDefun() local
81 .Finalize(&graph, &map_defun_node)); in WrapFunctionWithMapDefun()
83 for (size_t i = 0; i < map_defun_node->num_outputs(); ++i) { in WrapFunctionWithMapDefun()
86 .Input(map_defun_node, i) in WrapFunctionWithMapDefun()
100 const NodeDef& map_defun_node = outer.node_def(0); in WrapAndVectorize() local
105 TF_RETURN_IF_ERROR(VectorizeMapDefun(outer, map_defun_node, lib, result)); in WrapAndVectorize()
252 auto map_defun_node = vectorized->node_def( in TEST() local
264 lib_def.Find(map_defun_node.attr().at("f").func().name()); in TEST()
477 const NodeDef& map_defun_node = vectorized->node_def(0); in TEST() local
478 EXPECT_EQ(map_defun_node.op(), "MapDefun"); in TEST()
[all …]
|
| D | vectorization_utils.h | 90 const NodeDef& map_defun_node, FunctionDefLibrary* lib,
|