Searched refs:shape_node (Results 1 – 5 of 5) sorted by relevance
206 Node* shape_node = nullptr; in ReplaceBroadcast() local210 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()
127 Node* shape_node; in rewriteNode() local131 .Finalize(g, &shape_node)); in rewriteNode()138 .Input(shape_node) in rewriteNode()
447 Node* shape_node; in StackTensor() local449 node_builder("Shape").Input(input_node).Finalize(g, &shape_node)); in StackTensor()458 .Input(shape_node) // input in StackTensor()
508 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape() argument510 if (shape_node.op() == "Shape") { in ExtractShape()512 properties.GetInputProperties(shape_node.name()); in ExtractShape()525 if (shape_node.attr().count("value") == 0) { in ExtractShape()528 const TensorProto& raw_val = shape_node.attr().at("value").tensor(); in ExtractShape()2583 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local2585 shape_node) in SimplifyReduction()2589 shape_node->set_device(node->device()); in SimplifyReduction()2590 node_map_->AddNode(shape_node->name(), shape_node); in SimplifyReduction()2592 shape_node->add_input(AsControlDependency(reduction_indices_input)); in SimplifyReduction()[all …]
1388 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local1390 graph->AddControlEdge(graph->source_node(), shape_node); in TEST_F()