Searched refs:variable_start_index (Results 1 – 4 of 4) sorted by relevance
161 int variable_start_index = num_args; in RewriteSubgraph() local166 variable_start_index == num_args) { in RewriteSubgraph()167 variable_start_index = i; in RewriteSubgraph()172 VLOG(4) << "variable_start_index: " << variable_start_index; in RewriteSubgraph()184 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()253 const int variable_start_index = xla_function_info.variable_start_index; in BuildXlaLaunchOps() local254 const int num_variables = num_inputs - variable_start_index; in BuildXlaLaunchOps()255 const int num_args = variable_start_index; in BuildXlaLaunchOps()278 int pos = variable_start_index + i; in BuildXlaLaunchOps()373 &result.variable_start_index)); in BuildXlaLaunchOps()
60 int variable_start_index = -1; member
888 int variable_start_index = 0; in BuildXlaLaunchOps() local889 for (; variable_start_index < tin_dtypes.size(); ++variable_start_index) { in BuildXlaLaunchOps()890 if (tin_dtypes.at(variable_start_index) == DT_RESOURCE) break; in BuildXlaLaunchOps()892 result.variable_start_index = variable_start_index; in BuildXlaLaunchOps()
198 int variable_start_index = num_args; in RewriteSubgraph() local204 !is_replicated_input(*args[i]) && variable_start_index == num_args) { in RewriteSubgraph()205 variable_start_index = i; in RewriteSubgraph()213 VLOG(4) << "variable_start_index: " << variable_start_index in RewriteSubgraph()226 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()2556 int variable_start_index; in BuildTPUReplicateOps() local2558 &variable_start_index)); in BuildTPUReplicateOps()2626 std::max(0, guaranteed_const_start_index - variable_start_index); in BuildTPUReplicateOps()