Home
last modified time | relevance | path

Searched refs:size_node (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc114 NodeDef size_node(*fanout.node); in Optimize() local
116 size_node.set_op("Size"); in Optimize()
117 size_node.set_input(0, node.input(0)); in Optimize()
118 size_node.set_input(1, AsControlDependency(node)); in Optimize()
119 size_node.mutable_attr()->erase("Tidx"); in Optimize()
120 size_node.mutable_attr()->erase("keep_dims"); in Optimize()
121 (*size_node.mutable_attr())["out_type"] = fanout.node->attr().at("T"); in Optimize()
122 (*size_node.mutable_attr())["T"].set_type(type); in Optimize()
127 size_node.set_device(node.device()); in Optimize()
131 Status s = IsKernelRegisteredForNode(size_node); in Optimize()
[all …]
Dgeneric_layout_optimizer_transposer_test.cc2508 auto* size_node = context.graph_view->GetNode( in TEST_F() local
2510 ASSERT_NE(size_node, nullptr); in TEST_F()
2524 VerifyRegularFaninMatch(updated_maxpoolgrad, 3, size_node->GetName(), 0); in TEST_F()
3331 auto* size_node = context.graph_view->GetNode( in TEST_F() local
3333 ASSERT_NE(size_node, nullptr); in TEST_F()
3334 ASSERT_EQ(size_node->NumRegularFanins(), 1); in TEST_F()
3335 VerifyRegularFaninMatch(size_node, 0, "size", 0); in TEST_F()
3343 VerifyRegularFaninMatch(updated_slice_node, 2, size_node->GetName(), 0); in TEST_F()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports_test.cc111 NodeDef* size_node = graph_def.add_node(); in TEST() local
112 size_node->set_name("size_node"); in TEST()
113 size_node->set_op("Const"); in TEST()
116 SetNodeTensorAttr<float>("value", size_tensor, size_node); in TEST()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc993 Node* size_node = graph->AddNode(size_def, &s); in CreateSliceNode() local
1006 slice_def.add_input(absl::StrCat(size_node->name(), ":0")); in CreateSliceNode()
1012 graph->AddEdge(size_node, 0, slice_node, 2); in CreateSliceNode()