| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_placer_test.cc | 29 DeviceProperties cpu_device; in TEST() local
59 DeviceProperties cpu_device; in TEST() local
89 DeviceProperties cpu_device; in TEST() local
117 DeviceProperties cpu_device; in TEST() local
142 DeviceProperties cpu_device; in TEST() local
170 DeviceProperties cpu_device; in TEST() local
193 DeviceProperties cpu_device; in TEST() local
270 DeviceProperties cpu_device; in TEST() local
287 DeviceProperties cpu_device; in TEST() local
|
| D | analytical_cost_estimator_test.cc | 32 DeviceProperties cpu_device; in SetUp() local
|
| D | virtual_scheduler_test.cc | 678 DeviceProperties cpu_device = GetDummyCPUDevice(); in VirtualSchedulerTest() local
695 DeviceProperties cpu_device; in GetDummyCPUDevice() local
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils_test.cc | 60 auto cpu_device = std::make_unique<util::SimpleDevice>(); in TEST() local
96 auto cpu_device = std::make_unique<util::SimpleDevice>(); in TEST() local
159 auto cpu_device = std::make_unique<util::SimpleDevice>(); in TEST() local
|
| D | eval_utils.cc | 78 LogicalResult EvaluateOperation(tensorflow::DeviceBase *cpu_device, in EvaluateOperation()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 136 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
211 Device* cpu_device; in Instantiate() local
|
| D | batch_kernels.cc | 378 Device* cpu_device; in InstantiateFunction() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding_test.cc | 92 DummyDevice cpu_device(Env::Default()); in Validate() local
180 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
|
| /external/mesa3d/src/vulkan/device-select-layer/ |
| D | device_select.h | 41 bool cpu_device; member
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | tf_optimizer_wrapper.cc | 96 tensorflow::DeviceBase* cpu_device = nullptr; in PYBIND11_MODULE() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
|
| D | loop_optimizer.cc | 557 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
725 DeviceBase* cpu_device) in LoopOptimizer()
|
| D | static_schedule_test.cc | 35 DeviceProperties cpu_device; in CreateVirtualCluster() local
|
| D | meta_optimizer.cc | 265 MetaOptimizer::MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg) in MetaOptimizer()
1358 DeviceBase* cpu_device, Cluster* cluster, in RunMetaOptimizer()
1370 Device* cpu_device, const ConfigProto& config_proto, in OptimizeGraph()
|
| D | memory_optimizer_test.cc | 207 DeviceProperties cpu_device; in CreateVirtualCluster() local
|
| /external/tensorflow/tensorflow/core/tfrt/runtime/ |
| D | runtime.cc | 84 auto cpu_device = corert->GetHostContext()->GetHostDeviceRef(); in InitializeOpHandlers() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster_test.cc | 36 DeviceProperties cpu_device; in SetUp() local
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tfrt_graph_execution_state.cc | 257 const Device* cpu_device, Graph* graph) { in PlaceInputOutputNodesOnHost()
300 const Device* cpu_device, Graph* graph) { in AdjustDeviceAssignment()
372 const Device* cpu_device = fallback_state.device_manager().HostCPU(); in BuildXlaOpsAndMaybeInsertTransferOps() local
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 142 Device* cpu_device; in TransformGraphFunction() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 62 Device* cpu_device() { return cpu_device_; } in cpu_device() function in tensorflow::__anond84d0c860111::TestEnv
|
| /external/pytorch/test/ |
| D | test_xpu.py | 33 cpu_device = torch.device("cpu") variable
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 247 Device* cpu_device = devices[0].get(); in RuntimeGraphOptimizer() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | trt_convert_api.cc | 86 tensorflow::DeviceBase* cpu_device = nullptr; in RunGrappler() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 135 string cpu_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in MergeOutgoingDataEdges() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | core_test.py | 1072 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest
|