Home
last modified time | relevance | path

Searched defs: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
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.cc671 DeviceProperties cpu_device = GetDummyCPUDevice(); in VirtualSchedulerTest() local
688 DeviceProperties cpu_device; in GetDummyCPUDevice() local
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc135 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
210 Device* cpu_device; in Instantiate() local
Dbatch_kernels.cc335 Device* cpu_device; in InstantiateFunction() local
/external/mesa3d/src/vulkan/device-select-layer/
Ddevice_select.h33 bool cpu_device; member
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc92 DummyDevice cpu_device(Env::Default()); in Validate() local
180 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
/external/tensorflow/tensorflow/python/grappler/
Dtf_optimizer_wrapper.cc94 tensorflow::DeviceBase* cpu_device = nullptr; in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
Dmeta_optimizer.cc222 MetaOptimizer::MetaOptimizer(DeviceBase* cpu_device, const ConfigProto& cfg) in MetaOptimizer()
964 DeviceBase* cpu_device, Cluster* cluster, in RunMetaOptimizer()
976 Device* cpu_device, const ConfigProto& config_proto, in OptimizeGraph()
Dloop_optimizer.cc557 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
725 DeviceBase* cpu_device) in LoopOptimizer()
Dstatic_schedule_test.cc35 DeviceProperties cpu_device; in CreateVirtualCluster() local
Dmemory_optimizer_test.cc207 DeviceProperties cpu_device; in CreateVirtualCluster() local
Dgeneric_layout_optimizer_test.cc171 DeviceProperties cpu_device; in SetUp() local
Dconstant_folding.cc190 DeviceBase* cpu_device, in ConstantFolding()
201 ConstantFolding::ConstantFolding(DeviceBase* cpu_device, in ConstantFolding()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc36 DeviceProperties cpu_device; in SetUp() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc171 Device* cpu_device = device_mgr->HostCPU(); in RunGrappler() local
/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
Dexecute.cc270 Device* cpu_device = ctx.HostCPU(); in GetDeviceForInput() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc247 Device* cpu_device = devices[0].get(); in RuntimeGraphOptimizer() local
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc135 string cpu_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in MergeOutgoingDataEdges() local
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py1062 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc627 Device* cpu_device; in Instantiate() local
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc817 Device* cpu_device; in InstantiateMultiDevice() local
Dplacer_test.cc1562 const string cpu_device = "/job:b/replica:0/task:0/device:FakeCPU:0"; in TEST_F() local

12