Home
last modified time | relevance | path

Searched refs:cpu_device (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc29 DeviceProperties cpu_device; in TEST() local
30 cpu_device.set_type("CPU"); in TEST()
31 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in TEST()
59 DeviceProperties cpu_device; in TEST() local
60 cpu_device.set_type("CPU"); in TEST()
61 devices["/CPU:0"] = cpu_device; in TEST()
89 DeviceProperties cpu_device; in TEST() local
90 cpu_device.set_type("CPU"); in TEST()
91 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in TEST()
117 DeviceProperties cpu_device; in TEST() local
[all …]
Danalytical_cost_estimator_test.cc32 DeviceProperties cpu_device; in SetUp() local
33 cpu_device.set_type("CPU"); in SetUp()
34 cpu_device.set_num_cores(4); in SetUp()
35 cpu_device.set_frequency(2600); in SetUp()
36 cpu_device.set_bandwidth(24 * 1024 * 1024); in SetUp()
37 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in SetUp()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc36 DeviceProperties cpu_device; in SetUp() local
37 cpu_device.set_type("CPU"); in SetUp()
38 cpu_device.set_frequency(1000); in SetUp()
39 cpu_device.set_num_cores(4); in SetUp()
40 cpu_device.set_bandwidth(32); in SetUp()
41 cpu_device.set_l1_cache_size(32 * 1024); in SetUp()
42 cpu_device.set_l2_cache_size(256 * 1024); in SetUp()
43 cpu_device.set_l3_cache_size(4 * 1024 * 1024); in SetUp()
44 cpu_device.set_memory_size(1024 * 1024); in SetUp()
46 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in SetUp()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode() argument
66 if (cpu_device == nullptr) { in EvaluateNode()
68 cpu_device = device.get(); in EvaluateNode()
72 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
76 params.device = cpu_device; in EvaluateNode()
Dstatic_schedule_test.cc35 DeviceProperties cpu_device; in CreateVirtualCluster() local
36 cpu_device.set_type("CPU"); in CreateVirtualCluster()
37 cpu_device.set_frequency(1000); in CreateVirtualCluster()
38 cpu_device.set_num_cores(4); in CreateVirtualCluster()
39 cpu_device.set_bandwidth(32); in CreateVirtualCluster()
40 cpu_device.set_l1_cache_size(32 * 1024); in CreateVirtualCluster()
41 cpu_device.set_l2_cache_size(256 * 1024); in CreateVirtualCluster()
42 cpu_device.set_l3_cache_size(4 * 1024 * 1024); in CreateVirtualCluster()
44 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in CreateVirtualCluster()
Dmeta_optimizer.h37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
124 DeviceBase* cpu_device, Cluster* cluster,
148 Device* cpu_device, const ConfigProto& config_proto,
Dgeneric_layout_optimizer_test.cc171 DeviceProperties cpu_device; in SetUp() local
172 cpu_device.set_type("CPU"); in SetUp()
173 cpu_device.set_frequency(1000); in SetUp()
174 cpu_device.set_num_cores(4); in SetUp()
175 cpu_device.set_bandwidth(32); in SetUp()
176 cpu_device.set_l1_cache_size(32 * 1024); in SetUp()
177 cpu_device.set_l2_cache_size(256 * 1024); in SetUp()
178 cpu_device.set_l3_cache_size(4 * 1024 * 1024); in SetUp()
179 cpu_device.set_memory_size(1024 * 1024); in SetUp()
185 absl::WrapUnique(new VirtualCluster({{"/CPU:0", cpu_device}, in SetUp()
[all …]
Devaluation_utils.h55 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
Dmemory_optimizer_test.cc207 DeviceProperties cpu_device; in CreateVirtualCluster() local
208 cpu_device.set_type("CPU"); in CreateVirtualCluster()
209 cpu_device.set_frequency(1000); in CreateVirtualCluster()
210 cpu_device.set_num_cores(4); in CreateVirtualCluster()
211 cpu_device.set_bandwidth(32); in CreateVirtualCluster()
212 cpu_device.set_memory_size(1024 * 1024); in CreateVirtualCluster()
221 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in CreateVirtualCluster()
Dloop_optimizer.h37 DeviceBase* cpu_device);
Dmeta_optimizer.cc222 MetaOptimizer::MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg) in MetaOptimizer() argument
223 : cpu_device_(cpu_device), in MetaOptimizer()
964 DeviceBase* cpu_device, Cluster* cluster, in RunMetaOptimizer() argument
966 MetaOptimizer optimizer(cpu_device, cfg); in RunMetaOptimizer()
976 Device* cpu_device, const ConfigProto& config_proto, in OptimizeGraph() argument
1013 std::move(item), config_proto, cpu_device, &cluster, &out_graph)); in OptimizeGraph()
Dloop_optimizer.cc557 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands() argument
576 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs)); in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout() argument
703 *switch_ctrl_node, *operand_0, *operand_1, cpu_device, resource_mgr, in CheckForDeadFanout()
725 DeviceBase* cpu_device) in LoopOptimizer() argument
727 cpu_device_(cpu_device), in LoopOptimizer()
Dconstant_folding.h48 explicit ConstantFolding(DeviceBase* cpu_device,
51 ConstantFolding(RewriterConfig::Toggle opt_level, DeviceBase* cpu_device,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc92 DummyDevice cpu_device(Env::Default()); in Validate() local
93 response.InitAlloc(&cpu_device, AllocatorAttributes()); in Validate()
180 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
184 response.InitAlloc(&cpu_device, AllocatorAttributes()); in BM_TensorResponse()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc62 Device* cpu_device() { return cpu_device_; } in cpu_device() function in tensorflow::__anond3bba5f70111::TestEnv
117 nullptr, env.cpu_device()); in BM_KernelAndDeviceInit()
138 nullptr, env.cpu_device()); in BM_KernelAndDeviceRun()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc135 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices() argument
210 Device* cpu_device; in Instantiate() local
211 TF_RETURN_IF_ERROR(lib->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in Instantiate()
226 FillOutputDevices(*lib, *cpu_device, AttrSlice(&func_->attr()), &opts)); in Instantiate()
Dpartitioned_function_ops.h45 const Device& cpu_device, AttrSlice attrs,
Dbatch_kernels.cc335 Device* cpu_device; in InstantiateFunction() local
336 TF_RETURN_IF_ERROR(flib_->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in InstantiateFunction()
353 opts.input_devices.push_back(cpu_device->name()); in InstantiateFunction()
363 opts.input_devices.push_back(cpu_device->name()); in InstantiateFunction()
369 opts.output_devices.push_back(cpu_device->name()); in InstantiateFunction()
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_wrapper.cc94 tensorflow::DeviceBase* cpu_device = nullptr; in PYBIND11_MODULE() local
96 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py116 cpu_device = tf_device.DeviceSpec.from_string(device)
117 cpu_device = cpu_device.replace(device_type="CPU", device_index=0)
118 return cpu_device.to_string()
Dtest_util_test.py82 cpu_device = config.list_physical_devices('CPU')[0]
83 self.assertLen(config.get_logical_device_configuration(cpu_device), 3)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc171 Device* cpu_device = device_mgr->HostCPU(); in RunGrappler() local
175 std::move(*item), config_proto, cpu_device, &cluster, &output_graph_def)); in RunGrappler()
/external/mesa3d/src/vulkan/device-select-layer/
Ddevice_select.h33 bool cpu_device; member
Ddevice_select_layer.c253 drm_device->cpu_device = properties.properties.deviceType == VK_PHYSICAL_DEVICE_TYPE_CPU; in fill_drm_device_info()
263 return drm_device->cpu_device; in fill_drm_device_info()
335 if (!pci_infos[i].cpu_device){ in device_select_find_non_cpu()
350 if (pci_infos[i].cpu_device) in find_non_cpu_skip()
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py1062 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest
1077 device_name = self.cpu_device
1096 self._send(t0, 't0', self.cpu_device)
1097 self._send(t1, 't1', self.cpu_device)
1099 self._recv(dtypes.float32, 't0', self.cpu_device),
1102 self._recv(dtypes.float32, 't1', self.cpu_device),
1111 self._send(t0, 't0', self.cpu_device)
1119 self._recv(dtypes.float32, 't1', self.cpu_device),

12