| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 126 int core_index) { in GetExecutableAliasConfig() argument
128 tpu_program_group->executable_info(core_index); in GetExecutableAliasConfig()
132 tpu_program_group->hlo_metadata(core_index) in GetExecutableAliasConfig()
192 xla::Backend* const backend, se::Stream* stream, int core_index, in RunExecutable() argument
213 tpu_program_group->host_transfer_info(core_index); in RunExecutable()
217 *tpu_program_group->hlo_metadata(core_index), in RunExecutable()
222 tpu_program_group->tpu_program(core_index))); in RunExecutable()
234 xla::Backend* const backend, se::Stream* stream, int core_index, in ExecuteTPUProgram() argument
240 tpu_program_group, backend, stream, core_index, in ExecuteTPUProgram()
328 const int core_index = centry.core_index(); in DoWork() local
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_entry.h | 31 const TpuProgramGroupInterface* tpu_program_group, int core_index) in TpuCompilationCacheEntry() argument
32 : tpu_program_group_(tpu_program_group), core_index_(core_index) {} in TpuCompilationCacheEntry()
41 int core_index() const { return core_index_; } in core_index() function
|
| D | tpu_compilation_cache_rpc_support.cc | 87 CHECK_GE(cache_entry.core_index(), 0); in SerializeCacheEntryToBufferSlices()
88 CHECK_LT(cache_entry.core_index(), tpu_program_group->program_count()); in SerializeCacheEntryToBufferSlices()
101 cache_entry.core_index(), &executable); in SerializeCacheEntryToBufferSlices()
116 tpu_program_group->may_modify_variables(cache_entry.core_index()); in SerializeCacheEntryToBufferSlices()
126 tpu_program_group->SerializeCompilerMetadata(cache_entry.core_index(), in SerializeCacheEntryToBufferSlices()
|
| D | tpu_execute_op.cc | 652 const int core_index = entry.core_index(); in DoWork() local
654 tpu_program_group->executable_info(core_index); in DoWork()
753 tpu_program_group->host_transfer_info(core_index); in DoWork()
757 *tpu_program_group->hlo_metadata(core_index), std::move(input), in DoWork()
761 tpu_program_group->tpu_program(core_index))); in DoWork()
|
| /external/arm-trusted-firmware/plat/imx/imx8m/include/ |
| D | gpc.h | 61 void imx_set_cpu_secure_entry(unsigned int core_index, uintptr_t sec_entrypoint);
62 void imx_set_cpu_pwr_off(unsigned int core_index);
63 void imx_set_cpu_pwr_on(unsigned int core_index);
64 void imx_set_cpu_lpm(unsigned int core_index, bool pdn);
|
| /external/cpuinfo/src/x86/linux/ |
| D | init.c | 349 …uint32_t processor_index = UINT32_MAX, core_index = UINT32_MAX, cluster_index = UINT32_MAX, packag… in cpuinfo_x86_linux_init() local
364 core_index++; in cpuinfo_x86_linux_init()
384 processors[processor_index].core = cores + core_index; in cpuinfo_x86_linux_init()
392 cores[core_index] = (struct cpuinfo_core) { in cpuinfo_x86_linux_init()
407 cores[core_index].processor_count++; in cpuinfo_x86_linux_init()
414 clusters[cluster_index].core_start = core_index; in cpuinfo_x86_linux_init()
431 packages[package_index].core_start = core_index; in cpuinfo_x86_linux_init()
441 linux_cpu_to_core_map[x86_linux_processors[i].linux_id] = cores + core_index; in cpuinfo_x86_linux_init()
|
| /external/cpuinfo/src/arm/mach/ |
| D | init.c | 86 …uarch decode_uarch(uint32_t cpu_family, uint32_t cpu_subtype, uint32_t core_index, uint32_t core_c… in decode_uarch() argument
100 return core_index < 2 ? cpuinfo_uarch_monsoon : cpuinfo_uarch_mistral; in decode_uarch()
103 return core_index + 4 < core_count ? cpuinfo_uarch_vortex : cpuinfo_uarch_tempest; in decode_uarch()
106 return core_index + 4 < core_count ? cpuinfo_uarch_lightning : cpuinfo_uarch_thunder; in decode_uarch()
109 return core_index + 4 < core_count ? cpuinfo_uarch_firestorm : cpuinfo_uarch_icestorm; in decode_uarch()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.h | 241 std::string GetDeviceAliasForLogicalCore(int core_index);
|
| D | tpu_rewrite_device_util.cc | 485 std::string GetDeviceAliasForLogicalCore(int core_index) { in GetDeviceAliasForLogicalCore() argument
486 return llvm::formatv("{0}_{1}", kTPUReplicatedCore, core_index).str(); in GetDeviceAliasForLogicalCore()
|