Searched refs:dynamic_shape_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 343 std::vector<Node*>* dynamic_shape_nodes);
364 const std::vector<Node*>& dynamic_shape_nodes, Graph* graph,
|
| D | distributed_tpu_rewrite_pass.cc | 2236 Graph* graph, std::vector<Node*>* dynamic_shape_nodes) { in BuildDynamicShapeNodes() argument
2237 dynamic_shape_nodes->clear(); in BuildDynamicShapeNodes()
2293 dynamic_shape_nodes->push_back(shape_node); in BuildDynamicShapeNodes()
2317 const std::vector<Node*>& dynamic_shape_nodes, Graph* graph, in BuildCompileNode() argument
2435 AddNodeAttr("NumDynamicShapes", static_cast<int>(dynamic_shape_nodes.size()), in BuildCompileNode()
2446 for (int i = 0; i < dynamic_shape_nodes.size(); ++i) { in BuildCompileNode()
2447 graph->AddEdge(dynamic_shape_nodes[i], 0, *compile_node, i); in BuildCompileNode()
2452 dynamic_shape_nodes.size() + i); in BuildCompileNode()
4314 std::vector<Node*> dynamic_shape_nodes; in RewriteTPUReplicateNode() local
4317 &dynamic_shape_nodes)); in RewriteTPUReplicateNode()
[all …]
|