Searched refs:variable_indices (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 81 OpKernelContext* ctx, absl::Span<const int> variable_indices, in GetVariableInfosFromCtxInputs() argument
85 variable_indices, std::back_inserter(resource_handles), in GetVariableInfosFromCtxInputs()
92 result->reserve(variable_indices.size()); in GetVariableInfosFromCtxInputs()
93 for (int i = 0; i < variable_indices.size(); i++) { in GetVariableInfosFromCtxInputs()
97 result->emplace_back(variable_indices[i], variable); in GetVariableInfosFromCtxInputs()
148 absl::Span<const int> variable_indices, in SnapshotResourceVariables() argument
152 GetVariableInfosFromCtxInputs(ctx, variable_indices, &variable_infos)); in SnapshotResourceVariables()
155 for (int i = 0; i < variable_indices.size(); i++) { in SnapshotResourceVariables()
157 OptionalTensor& tensor = (*result)[variable_indices[i]]; in SnapshotResourceVariables()
158 tensor.name = HandleFromInput(ctx, variable_indices[i]).name(); in SnapshotResourceVariables()
[all …]
|
| D | xla_launch_util.h | 59 absl::Span<const int> variable_indices,
|