Searched refs:variable_start_index (Results 1 – 2 of 2) sorted by relevance
156 int variable_start_index = num_args; in RewriteSubgraph() local161 variable_start_index == num_args) { in RewriteSubgraph()162 variable_start_index = i; in RewriteSubgraph()167 VLOG(4) << "variable_start_index: " << variable_start_index; in RewriteSubgraph()179 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()252 int variable_start_index; in BuildXlaLaunchOps() local254 &variable_start_index)); in BuildXlaLaunchOps()260 const int num_variables = num_inputs - variable_start_index; in BuildXlaLaunchOps()261 const int num_args = variable_start_index; in BuildXlaLaunchOps()284 int pos = variable_start_index + i; in BuildXlaLaunchOps()
214 int variable_start_index = num_args; in RewriteSubgraph() local220 !is_replicated_input(*args[i]) && variable_start_index == num_args) { in RewriteSubgraph()221 variable_start_index = i; in RewriteSubgraph()229 VLOG(4) << "variable_start_index: " << variable_start_index in RewriteSubgraph()242 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()2604 int variable_start_index; in BuildTPUReplicateOps() local2606 &variable_start_index)); in BuildTPUReplicateOps()2674 std::max(0, guaranteed_const_start_index - variable_start_index); in BuildTPUReplicateOps()