Home
last modified time | relevance | path

Searched refs:shape_node (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc206 Node* shape_node = nullptr; in ReplaceBroadcast() local
210 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()
211 shape_node->set_assigned_device_name_index(send_dev); in ReplaceBroadcast()
225 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()
226 shape_node->set_assigned_device_name_index(recv_dev); in ReplaceBroadcast()
231 .Input(shape_node) in ReplaceBroadcast()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dreshape_builder.cc86 auto* shape_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
90 AddInput(TensorID(shape_node->GetID(), 0)); in PopulateSubGraph()
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc159 Node* shape_node; in RewriteIntoTempVariable() local
163 .Finalize(g, &shape_node)); in RewriteIntoTempVariable()
170 .Input(shape_node) in RewriteIntoTempVariable()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc485 auto* shape_node = graph_view.GetNode("shape"); in TEST_F() local
486 ASSERT_NE(shape_node, nullptr); in TEST_F()
487 ASSERT_EQ(shape_node->NumRegularFanins(), 1); in TEST_F()
488 VerifyRegularFaninMatch(shape_node, 0, conv2d_node->GetName(), 0); in TEST_F()
493 VerifyRegularFaninMatch(fill_node, 0, shape_node->GetName(), 0); in TEST_F()
Dconstant_folding.cc562 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape() argument
564 if (shape_node.op() == "Shape") { in ExtractShape()
566 properties.GetInputProperties(shape_node.name()); in ExtractShape()
579 if (shape_node.attr().count("value") == 0) { in ExtractShape()
582 const TensorProto& raw_val = shape_node.attr().at("value").tensor(); in ExtractShape()
2878 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local
2880 shape_node) in SimplifyReduction()
2884 shape_node->set_device(node->device()); in SimplifyReduction()
2885 node_map_->AddNode(shape_node->name(), shape_node); in SimplifyReduction()
2887 shape_node->add_input(AsControlDependency(reduction_indices_input)); in SimplifyReduction()
[all …]
Dgeneric_layout_optimizer_transposer_test.cc3092 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local
3093 ASSERT_NE(shape_node, nullptr); in TEST_F()
3094 TF_ASSERT_OK(shape_transposer.TransposeNode(&context, shape_node)); in TEST_F()
3161 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local
3162 ASSERT_NE(shape_node, nullptr); in TEST_F()
3163 TF_ASSERT_OK(shape_transposer.TransposeNode(&context, shape_node)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc440 Node* shape_node; in StackTensor() local
442 node_builder("Shape").Input(input_node).Finalize(g, &shape_node)); in StackTensor()
451 .Input(shape_node) // input in StackTensor()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2291 Node* shape_node = graph->AddNode(def, &status); in BuildDynamicShapeNodes() local
2293 dynamic_shape_nodes->push_back(shape_node); in BuildDynamicShapeNodes()
2295 shape_node->set_assigned_device_name(src->assigned_device_name()); in BuildDynamicShapeNodes()
2296 graph->AddEdge(src, src_output, shape_node, 0); in BuildDynamicShapeNodes()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1456 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local
1458 graph->AddControlEdge(graph->source_node(), shape_node); in TEST_F()