Home
last modified time | relevance | path

Searched defs:cpu_device (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc29 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
Danalytical_cost_estimator_test.cc32 DeviceProperties cpu_device; in SetUp() local
Dvirtual_scheduler_test.cc66 DeviceProperties cpu_device = GetDummyCPUDevice(); in VirtualSchedulerTest() local
95 DeviceProperties cpu_device; in GetDummyCPUDevice() local
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc129 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
185 Device* cpu_device; in Instantiate() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc92 DummyDevice cpu_device(Env::Default()); in Validate() local
179 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer.i110 tensorflow::DeviceBase* cpu_device = nullptr; in TF_OptimizeGraph() local
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc35 DeviceProperties cpu_device; in SetUp() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmeta_optimizer.cc148 MetaOptimizer::MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg) in MetaOptimizer()
702 DeviceBase* cpu_device, Cluster* cluster, in RunMetaOptimizer()
717 Device* cpu_device, const ConfigProto& config_proto, in OptimizeGraph()
Dstatic_schedule_test.cc34 DeviceProperties cpu_device; in CreateVirtualCluster() local
Devaluation_utils.cc83 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
Dloop_optimizer.cc558 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands()
598 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
703 DeviceBase* cpu_device) in LoopOptimizer()
Dmemory_optimizer_test.cc207 DeviceProperties cpu_device; in CreateVirtualCluster() local
Dconstant_folding.cc228 DeviceBase* cpu_device) in ConstantFolding()
233 ConstantFolding::ConstantFolding(DeviceBase* cpu_device) in ConstantFolding()
/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
Dexecute.cc261 tensorflow::Device** cpu_device) { in CPUDeviceOnTask()
325 Device* cpu_device = ctx->HostCPU(); in AddInputDevicesToCacheKey() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc113 Device* cpu_device = devices[0].get(); in OptimizeGraph() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc683 Device* cpu_device = nullptr; in OptimizeGraph() local
Dprocess_function_library_runtime.cc622 Device* cpu_device; in InstantiateMultiDevice() local
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py705 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest