Home
last modified time | relevance | path

Searched defs:core_index (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compilation_cache_entry.h31 const TpuProgramGroupInterface* tpu_program_group, int core_index) in TpuCompilationCacheEntry()
41 int core_index() const { return core_index_; } in core_index() function
Dtpu_reshard_variables_op.cc155 const int core_index = entry.core_index(); in DoTpuExecute() local
Dtpu_functional_ops.cc1785 for (int core_index = device_ordinal; in ReplaceAndPartitionXLAShardingVariable() local
1832 for (int core_index = 0; core_index < num_cores_per_replica; core_index++) { in ReplaceAndPartitionXLAShardingVariable() local
1844 for (int core_index = 0; core_index < num_cores_per_replica; core_index++) { in ReplaceAndPartitionXLAShardingVariable() local
1889 for (int core_index = 0; core_index < num_cores_per_replica; core_index++) { in ReplaceAndPartitionXLAShardingVariable() local
Dtpu_execute_op.cc648 const int core_index = entry.core_index(); in DoWork() local
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc126 int core_index) { in GetExecutableAliasConfig()
193 xla::Backend* const backend, se::Stream* stream, int core_index, in RunExecutable()
235 xla::Backend* const backend, se::Stream* stream, int core_index, in ExecuteTPUProgram()
330 const int core_index = centry.core_index(); in DoWork() local
442 const int core_index = centry.core_index(); in DoWork() local
/external/cpuinfo/src/arm/mach/
Dinit.c86 …info_uarch decode_uarch(uint32_t cpu_family, uint32_t cpu_subtype, uint32_t core_index, uint32_t c… in decode_uarch()
/external/cpuinfo/src/x86/linux/
Dinit.c349 …uint32_t processor_index = UINT32_MAX, core_index = UINT32_MAX, cluster_index = UINT32_MAX, packag… in cpuinfo_x86_linux_init() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc485 std::string GetDeviceAliasForLogicalCore(int core_index) { in GetDeviceAliasForLogicalCore()