Home
last modified time | relevance | path

Searched refs:map_defun_fn (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc69 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()
75 map_defun_fn->ret_types.size())); in UpdateMapDefunAttrs()
78 Status AddMapDefunOutput(FunctionBody* map_defun_fn, Node* map_defun_node, in AddMapDefunOutput() argument
81 int index = map_defun_fn->ret_nodes.size(); in AddMapDefunOutput()
90 Node* ret_node = map_defun_fn->graph->AddNode(ret_node_def, &s); in AddMapDefunOutput()
93 map_defun_fn->graph->AddEdge(output.first, output.second, ret_node, 0); in AddMapDefunOutput()
94 map_defun_fn->ret_nodes.push_back(ret_node); in AddMapDefunOutput()
95 map_defun_fn->ret_types.push_back(type); in AddMapDefunOutput()
96 UpdateMapDefunAttrs(map_defun_fn, map_defun_node); in AddMapDefunOutput()
[all …]
Dvectorization_utils_test.cc263 const FunctionDef* map_defun_fn = in TEST() local
265 EXPECT_EQ(map_defun_fn->signature().output_arg_size(), 1); in TEST()
480 const FunctionDef* map_defun_fn = in TEST() local
483 const NodeDef& print_node = map_defun_fn->node_def( in TEST()
484 function_utils::FindFunctionNodeWithOp("Print", *map_defun_fn)); in TEST()
485 const NodeDef& cast_node = map_defun_fn->node_def( in TEST()
486 function_utils::FindFunctionNodeWithOp("Cast", *map_defun_fn)); in TEST()