Home
last modified time | relevance | path

Searched defs:shape_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc127 Node* shape_node; in rewriteNode() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc206 Node* shape_node = nullptr; in ReplaceBroadcast() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc447 Node* shape_node; in StackTensor() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc508 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape()
2583 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc1388 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local