Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A 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
H A Danalytical_cost_estimator_test.cc32 DeviceProperties cpu_device; in SetUp() local
H A Dvirtual_scheduler_test.cc678 DeviceProperties cpu_device = GetDummyCPUDevice(); in VirtualSchedulerTest() local
695 DeviceProperties cpu_device; in GetDummyCPUDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils_test.cc60 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
H A Deval_utils.cc78 LogicalResult EvaluateOperation(tensorflow::DeviceBase *cpu_device, in EvaluateOperation()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpartitioned_function_ops.cc136 const FunctionLibraryRuntime& lib, const Device& cpu_device, in FillOutputDevices()
211 Device* cpu_device; in Instantiate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state.cc257 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
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dtensor_coding_test.cc92 DummyDevice cpu_device(Env::Default()); in Validate() local
180 DummyDevice cpu_device(Env::Default()); in BM_TensorResponse() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dloop_optimizer.cc557 DeviceBase* cpu_device, in EvaluateBoolOpForConstantOperands()
600 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in CheckForDeadFanout()
725 DeviceBase* cpu_device) in LoopOptimizer()
H A Dmeta_optimizer.cc265 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()
H A Devaluation_utils.cc62 DeviceBase* cpu_device, ResourceMgr* resource_mgr, in EvaluateNode()
H A Dstatic_schedule_test.cc35 DeviceProperties cpu_device; in CreateVirtualCluster() local
H A Dmemory_optimizer_test.cc207 DeviceProperties cpu_device; in CreateVirtualCluster() local
H A Dgeneric_layout_optimizer_test.cc171 DeviceProperties cpu_device; in SetUp() local
/aosp_15_r20/external/mesa3d/src/vulkan/device-select-layer/
H A Ddevice_select.h41 bool cpu_device; member
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dtf_optimizer_wrapper.cc96 tensorflow::DeviceBase* cpu_device = nullptr; in PYBIND11_MODULE() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Druntime.cc84 auto cpu_device = corert->GetHostContext()->GetHostDeviceRef(); in InitializeOpHandlers() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dvirtual_cluster_test.cc36 DeviceProperties cpu_device; in SetUp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtransform_graph_function.cc142 Device* cpu_device; in TransformGraphFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device_test.cc62 Device* cpu_device() { return cpu_device_; } in cpu_device() function in tensorflow::__anon79d852060111::TestEnv
/aosp_15_r20/external/pytorch/test/
H A Dtest_xpu.py33 cpu_device = torch.device("cpu") variable
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item_builder.cc247 Device* cpu_device = devices[0].get(); in RuntimeGraphOptimizer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc86 tensorflow::DeviceBase* cpu_device = nullptr; in RunGrappler() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dbuild_xla_ops_pass.cc135 string cpu_device = "/job:localhost/replica:0/task:0/device:CPU:0"; in MergeOutgoingDataEdges() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcore_test.py1072 cpu_device = '/job:localhost/replica:0/task:0/device:CPU:0' variable in SendRecvTest

12