Home
last modified time | relevance | path

Searched refs:set_num_cores (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/clusters/
Dutils.cc55 device.set_num_cores(port::NumSchedulableCPUs()); in GetLocalCPUInfo()
94 device.set_num_cores(properties.multiProcessorCount); in GetLocalGPUInfo()
131 device.set_num_cores(properties.multiProcessorCount); in GetLocalGPUInfo()
Dvirtual_cluster_test.cc39 cpu_device.set_num_cores(4); in SetUp()
/external/tensorflow/tensorflow/core/grappler/costs/
Danalytical_cost_estimator_test.cc34 cpu_device.set_num_cores(4); in SetUp()
40 gpu_device.set_num_cores(12); in SetUp()
Dgraph_memory_test.cc33 devices_["/CPU:0"].set_num_cores(1); in GraphMemoryTest()
38 devices_["/GPU:0"].set_num_cores(1); in GraphMemoryTest()
Dvirtual_scheduler_test.cc691 cpu_device.set_num_cores(2); in GetDummyCPUDevice()
Dop_level_cost_estimator_test.cc80 device->set_num_cores(10); in SetCpuDevice()
/external/tensorflow/tensorflow/core/profiler/utils/
Ddevice_caps_utils.cc61 caps.set_num_cores(stat.IntOrUintValue()); in GetDeviceCaps()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.h50 void set_num_cores(int64 tpu_cores) { num_cores_ = tpu_cores; } in set_num_cores() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule_test.cc38 cpu_device.set_num_cores(4); in CreateVirtualCluster()
Dmemory_optimizer_test.cc210 cpu_device.set_num_cores(4); in CreateVirtualCluster()
216 gpu_device.set_num_cores(24); in CreateVirtualCluster()
Dgeneric_layout_optimizer_test.cc174 cpu_device.set_num_cores(4); in SetUp()
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_op_stats.cc61 cap.set_num_cores(stat.IntValue()); in GetDeviceCapFromXPlane()