Searched refs:guaranteed_constant_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 356 const std::vector<Node*>& guaranteed_constant_nodes,
539 std::vector<Node*>* guaranteed_constant_nodes,
|
| D | distributed_tpu_rewrite_pass.cc | 2309 const std::vector<Node*>& guaranteed_constant_nodes, in BuildCompileNode() argument
2355 const int num_guaranteed_constants = guaranteed_constant_nodes.size(); in BuildCompileNode()
2377 guaranteed_constant_nodes[i - guaranteed_const_start_index] in BuildCompileNode()
2451 graph->AddEdge(guaranteed_constant_nodes[i], 0, *compile_node, in BuildCompileNode()
4100 std::vector<Node*>* guaranteed_constant_nodes, in DealWithConstantsAndVariables() argument
4103 replicate_node, input_name_map, guaranteed_constant_nodes)); in DealWithConstantsAndVariables()
4305 std::vector<Node*> guaranteed_constant_nodes; in RewriteTPUReplicateNode() local
4310 &guaranteed_constant_nodes, &variable_reads)); in RewriteTPUReplicateNode()
4323 arg_types, guaranteed_constant_nodes, session_handle, arg_sharding, in RewriteTPUReplicateNode()
|