| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_program_group.cc | 98 bool may_modify_variables; in Initialize() local
100 &may_modify_variables); in Initialize()
101 may_modify_variables_array[i] = may_modify_variables; in Initialize()
200 const std::vector<bool>& may_modify_variables) { in set_may_modify_variables() argument
201 may_modify_variables_ = may_modify_variables; in set_may_modify_variables()
204 bool TpuProgramGroup::may_modify_variables(int index) const { in may_modify_variables() function in tensorflow::tpu::TpuProgramGroup
207 bool may_modify_variables; in may_modify_variables() local
209 &may_modify_variables); in may_modify_variables()
210 return may_modify_variables; in may_modify_variables()
|
| D | tpu_compilation_cache_rpc_support.cc | 115 bool may_modify_variables = in SerializeCacheEntryToBufferSlices() local
116 tpu_program_group->may_modify_variables(cache_entry.core_index()); in SerializeCacheEntryToBufferSlices()
117 header.set_may_modify_variables(may_modify_variables); in SerializeCacheEntryToBufferSlices()
|
| D | tpu_program_group.h | 125 void set_may_modify_variables(const std::vector<bool>& may_modify_variables);
126 bool may_modify_variables(int index) const override;
|
| D | tpu_program_group_interface.h | 68 virtual bool may_modify_variables(int index) const = 0;
|
| D | tpu_compilation_cache_interface.h | 113 std::vector<bool>* may_modify_variables,
202 std::vector<bool>* may_modify_variables,
|
| D | tpu_compilation_cache_interface.cc | 366 std::vector<bool>* may_modify_variables, in CompileIfKeyAbsent() argument
372 sharding_key, may_modify_variables, &removed_entries, hlo_metadatas, in CompileIfKeyAbsent()
404 std::vector<bool>* may_modify_variables, in CompileIfKeyAbsentHelper() argument
504 *may_modify_variables = entry->tpu_program_group->may_modify_variables_list(); in CompileIfKeyAbsentHelper()
|
| D | tpu_compile_op_support.cc | 203 std::vector<bool>* may_modify_variables, in AddVariableUpdatesToCores() argument
207 may_modify_variables->resize(metadata.num_cores_per_replica(), false); in AddVariableUpdatesToCores()
224 (*may_modify_variables)[core] = in AddVariableUpdatesToCores()
225 (*may_modify_variables)[core] || update.modified; in AddVariableUpdatesToCores()
|
| D | tpu_compile_op_common.cc | 690 std::vector<bool> may_modify_variables; in ComputeInternal() local
694 &may_modify_variables, &hlo_metadatas, in ComputeInternal()
766 (may_modify_variables.size() * static_cast<int>(!use_mlir_)) != in ComputeInternal()
771 may_modify_variables.size() * static_cast<int>(!use_mlir_), in ComputeInternal()
835 for (bool m : may_modify_variables) { in ComputeInternal()
838 for (int i = 0; i < may_modify_variables.size(); ++i) { in ComputeInternal()
|
| D | tpu_compilation_cache.proto | 30 bool may_modify_variables = 3; field
|
| D | tpu_compile_op_support.h | 127 std::vector<bool>* may_modify_variables,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | TPUCompile.pbtxt | 22 name: "may_modify_variables"
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 195 std::vector<bool> may_modify_variables; in Compute() local
201 &may_modify_variables, &hlo_metadata, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_TPUCompile.pbtxt | 19 'may_modify_variables' indicates whether variables may be modified.
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_ops_c_api.h | 360 const XLA_TpuProgram* tpu_program, bool* may_modify_variables);
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 53738 name: "may_modify_variables"
|