Home
last modified time | relevance | path

Searched refs:cpu_device (Results 1 – 25 of 26) 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()
Dvirtual_scheduler_test.cc66 DeviceProperties cpu_device = GetDummyCPUDevice(); in VirtualSchedulerTest() local
70 devices[kCPU0] = cpu_device; in VirtualSchedulerTest()
71 devices[kCPU1] = cpu_device; in VirtualSchedulerTest()
95 DeviceProperties cpu_device; in GetDummyCPUDevice() local
96 cpu_device.set_type("CPU"); in GetDummyCPUDevice()
97 cpu_device.set_frequency(4000); in GetDummyCPUDevice()
98 cpu_device.set_num_cores(2); in GetDummyCPUDevice()
99 cpu_device.set_bandwidth(2000000); in GetDummyCPUDevice()
100 return cpu_device; in GetDummyCPUDevice()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc35 DeviceProperties cpu_device; in SetUp() local
36 cpu_device.set_type("CPU"); in SetUp()
37 cpu_device.set_frequency(1000); in SetUp()
38 cpu_device.set_num_cores(4); in SetUp()
39 cpu_device.set_bandwidth(32); in SetUp()
40 cpu_device.set_l1_cache_size(32 * 1024); in SetUp()
41 cpu_device.set_l2_cache_size(256 * 1024); in SetUp()
42 cpu_device.set_l3_cache_size(4 * 1024 * 1024); in SetUp()
43 cpu_device.set_memory_size(1024 * 1024); in SetUp()
45 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in SetUp()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule_test.cc34 DeviceProperties cpu_device; in CreateVirtualCluster() local
35 cpu_device.set_type("CPU"); in CreateVirtualCluster()
36 cpu_device.set_frequency(1000); in CreateVirtualCluster()
37 cpu_device.set_num_cores(4); in CreateVirtualCluster()
38 cpu_device.set_bandwidth(32); in CreateVirtualCluster()
39 cpu_device.set_l1_cache_size(32 * 1024); in CreateVirtualCluster()
40 cpu_device.set_l2_cache_size(256 * 1024); in CreateVirtualCluster()
41 cpu_device.set_l3_cache_size(4 * 1024 * 1024); in CreateVirtualCluster()
43 devices["/job:localhost/replica:0/task:0/cpu:0"] = cpu_device; in CreateVirtualCluster()
Devaluation_utils.cc83 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode() argument
87 if (cpu_device == nullptr) { in EvaluateNode()
89 cpu_device = device.get(); in EvaluateNode()
93 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
97 params.device = cpu_device; in EvaluateNode()
Dmeta_optimizer.h37 MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg);
110 DeviceBase* cpu_device, Cluster* cluster,
134 Device* cpu_device, const ConfigProto& config_proto,
Dmeta_optimizer.cc148 MetaOptimizer::MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg) in MetaOptimizer() argument
149 : cpu_device_(cpu_device), in MetaOptimizer()
702 DeviceBase* cpu_device, Cluster* cluster, in RunMetaOptimizer() argument
704 MetaOptimizer optimizer(cpu_device, cfg); in RunMetaOptimizer()
717 Device* cpu_device, const ConfigProto& config_proto, in OptimizeGraph() argument
754 item, config_proto, cpu_device, &cluster, &out_graph)); in OptimizeGraph()
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.cc558 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands() argument
574 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs)); in EvaluateBoolOpForConstantOperands()
598 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout() argument
686 *switch_ctrl_node, *operand_0, *operand_1, cpu_device, resource_mgr, in CheckForDeadFanout()
703 DeviceBase* cpu_device) in LoopOptimizer() argument
705 cpu_device_(cpu_device), in LoopOptimizer()
Devaluation_utils.h56 DeviceBase* cpu_device, ResourceMgr* resource_mgr,
Dconstant_folding.h45 explicit ConstantFolding(DeviceBase* cpu_device);
46 ConstantFolding(RewriterConfig::Toggle opt_level, DeviceBase* cpu_device);
Dloop_optimizer.h35 DeviceBase* cpu_device);
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc129 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices() argument
144 opts->output_devices.push_back(cpu_device.name()); in FillOutputDevices()
185 Device* cpu_device; in Instantiate() local
186 TF_RETURN_IF_ERROR(lib->device_mgr()->LookupDevice("CPU:0", &cpu_device)); in Instantiate()
196 opts.input_devices.push_back(cpu_device->name()); in Instantiate()
203 FillOutputDevices(*lib, *cpu_device, AttrSlice(&func_->attr()), &opts)); in Instantiate()
Dpartitioned_function_ops.h45 const Device& cpu_device, AttrSlice attrs,
/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()
179 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
183 response.InitAlloc(&cpu_device, AllocatorAttributes()); in BM_TensorResponse()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device_test.cc60 Device* cpu_device() { return cpu_device_; } in cpu_device() function in tensorflow::__anone5cc3cd90111::TestEnv
120 nullptr, env.cpu_device()); in BM_KernelAndDeviceInit()
143 nullptr, env.cpu_device()); in BM_KernelAndDeviceRun()
Dexecute.cc261 tensorflow::Device** cpu_device) { in CPUDeviceOnTask() argument
266 return ctx->FindDeviceByName(cpu_device_name, cpu_device); in CPUDeviceOnTask()
325 Device* cpu_device = ctx->HostCPU(); in AddInputDevicesToCacheKey() local
341 input_dev_ptrs->push_back(cpu_device); in AddInputDevicesToCacheKey()
344 device_name = device != nullptr ? device->name() : cpu_device->name(); in AddInputDevicesToCacheKey()
345 input_dev_ptrs->push_back(device == nullptr ? cpu_device : device); in AddInputDevicesToCacheKey()
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py705 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest
720 device_name = self.cpu_device
733 self._send(t0, 't0', self.cpu_device)
734 self._send(t1, 't1', self.cpu_device)
736 self._recv(dtypes.float32, 't0', self.cpu_device),
739 self._recv(dtypes.float32, 't1', self.cpu_device),
748 self._send(t0, 't0', self.cpu_device)
756 self._recv(dtypes.float32, 't1', self.cpu_device),
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.i110 tensorflow::DeviceBase* cpu_device = nullptr; in TF_OptimizeGraph() local
112 tensorflow::grappler::MetaOptimizer optimizer(cpu_device, config_proto); in TF_OptimizeGraph()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc113 Device* cpu_device = devices[0].get(); in OptimizeGraph() local
133 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in OptimizeGraph()
146 optimizer.Optimize(flr, env, cpu_device, &graphptr, /*shape_map=*/nullptr); in OptimizeGraph()
/external/v8/tools/
Drun_perf.py861 cpu_device = CustomMachineConfiguration.GetCPUPathForId(cpu_index)
862 with open(cpu_device, "r") as f:
880 cpu_device = CustomMachineConfiguration.GetCPUPathForId(cpu_index)
881 with open(cpu_device, "w") as f:
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc683 Device* cpu_device = nullptr; in OptimizeGraph() local
688 cpu_device = device; in OptimizeGraph()
694 item, session_options_->config, cpu_device, &cluster, &new_graph)); in OptimizeGraph()
Dprocess_function_library_runtime.cc622 Device* cpu_device; in InstantiateMultiDevice() local
623 TF_RETURN_IF_ERROR(device_mgr_->LookupDevice("CPU:0", &cpu_device)); in InstantiateMultiDevice()
629 &data->overlay_lib_, device_set, cpu_device, &graph); in InstantiateMultiDevice()
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_strategy.py70 cpu_device = tf_device.DeviceSpec.from_string(device)
71 cpu_device.merge_from(tf_device.DeviceSpec(device_type="CPU", device_index=0))
72 return cpu_device.to_string()

12