| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_program_group.cc | 101 bool may_modify_variables; in Initialize() local
103 &may_modify_variables); in Initialize()
104 may_modify_variables_array[i] = may_modify_variables; in Initialize()
196 const std::vector<bool>& may_modify_variables) { in set_may_modify_variables() argument
197 may_modify_variables_ = may_modify_variables; in set_may_modify_variables()
200 bool TpuProgramGroup::may_modify_variables(int index) const { in may_modify_variables() function in tensorflow::tpu::TpuProgramGroup
203 bool may_modify_variables; in may_modify_variables() local
205 &may_modify_variables); in may_modify_variables()
206 return may_modify_variables; in may_modify_variables()
|
| H A D | tpu_compile_op_common.cc | 287 std::vector<bool> may_modify_variables; in ComputeInternal() local
291 &may_modify_variables, &hlo_metadatas, in ComputeInternal()
360 if (may_modify_variables.size() == 1) { in ComputeInternal()
361 may_modify_variables.resize(metadata_.num_cores_per_replica(), in ComputeInternal()
362 may_modify_variables[0]); in ComputeInternal()
367 (may_modify_variables.size() * static_cast<int>(!use_mlir_)) != in ComputeInternal()
372 may_modify_variables.size() * static_cast<int>(!use_mlir_), in ComputeInternal()
437 for (bool m : may_modify_variables) { in ComputeInternal()
440 for (int i = 0; i < may_modify_variables.size(); ++i) { in ComputeInternal()
|
| H A D | tpu_compilation_cache_rpc_support.cc | 116 bool may_modify_variables = in SerializeCacheEntryToBufferSlices() local
117 tpu_program_group->may_modify_variables(cache_entry.core_index()); in SerializeCacheEntryToBufferSlices()
118 header.set_may_modify_variables(may_modify_variables); in SerializeCacheEntryToBufferSlices()
|
| H A D | tpu_compilation_cache_interface.cc | 370 std::vector<bool>* may_modify_variables, in CompileIfKeyAbsent() argument
376 sharding_key, may_modify_variables, &removed_entries, hlo_metadatas, in CompileIfKeyAbsent()
408 std::vector<bool>* may_modify_variables, in CompileIfKeyAbsentHelper() argument
508 *may_modify_variables = entry->tpu_program_group->may_modify_variables_list(); in CompileIfKeyAbsentHelper()
|
| H A D | tpu_compile_op_support.cc | 205 std::vector<bool>* may_modify_variables, in AddVariableUpdatesToCores() argument
209 may_modify_variables->resize(metadata.num_cores_per_replica(), false); in AddVariableUpdatesToCores()
227 (*may_modify_variables)[core] = in AddVariableUpdatesToCores()
228 (*may_modify_variables)[core] || update.modified; in AddVariableUpdatesToCores()
|
| H A D | tpu_program_group.h | 122 void set_may_modify_variables(const std::vector<bool>& may_modify_variables);
123 bool may_modify_variables(int index) const override;
|
| H A D | tpu_compilation_cache_interface.h | 113 std::vector<bool>* may_modify_variables,
202 std::vector<bool>* may_modify_variables,
|
| H A D | tpu_program_group_interface.h | 69 virtual bool may_modify_variables(int index) const = 0;
|
| H A D | tpu_compilation_cache.proto | 30 bool may_modify_variables = 3; field
|
| H A D | tpu_compile_op_support.h | 127 std::vector<bool>* may_modify_variables,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | tpu_compile_ops.cc | 196 std::vector<bool> may_modify_variables; in Compute() local
202 &may_modify_variables, &hlo_metadata, in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| H A D | TPUCompile.pbtxt | 22 name: "may_modify_variables"
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_TPUCompile.pbtxt | 19 'may_modify_variables' indicates whether variables may be modified.
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_ops_c_api.h | 429 const XLA_TpuProgram* tpu_program, bool* may_modify_variables);
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | ops.pbtxt | 57513 name: "may_modify_variables"
|