Home
last modified time | relevance | path

Searched refs:guaranteed_const_start_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc215 int guaranteed_const_start_index = num_args; in RewriteSubgraph() local
223 guaranteed_const_start_index == num_args) { in RewriteSubgraph()
224 guaranteed_const_start_index = i; in RewriteSubgraph()
230 << " guaranteed_const_start_index: " << guaranteed_const_start_index; in RewriteSubgraph()
243 AddNodeAttr("_guaranteed_const_start_index", guaranteed_const_start_index, in RewriteSubgraph()
2607 int guaranteed_const_start_index; in BuildTPUReplicateOps() local
2610 &guaranteed_const_start_index)); in BuildTPUReplicateOps()
2674 std::max(0, guaranteed_const_start_index - variable_start_index); in BuildTPUReplicateOps()
2677 in_edges.size() - guaranteed_const_start_index; in BuildTPUReplicateOps()
Ddistributed_tpu_rewrite_pass.cc2356 const int guaranteed_const_start_index = num_args - num_guaranteed_constants; in BuildCompileNode() local
2375 if (i >= guaranteed_const_start_index) { in BuildCompileNode()
2377 guaranteed_constant_nodes[i - guaranteed_const_start_index] in BuildCompileNode()
2418 constant_arg_types.push_back(arg_types[guaranteed_const_start_index + i]); in BuildCompileNode()