Home
last modified time | relevance | path

Searched defs:kCPUDevice (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/jit/codegen/fuser/
Dinterface.h14 constexpr int kCPUDevice = -1; variable
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc215 const char* kCPUDevice = "CPU:0"; in TensorHandleCopyBetweenDevices() local
292 const char* kCPUDevice = "CPU:0"; in TensorHandleCopyBetweenDevicesError() local
345 const char* kCPUDevice = "CPU:0"; in TensorHandleCopyBetweenTwoGPUDevices() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass_test.cc49 const char kCPUDevice[] = "/job:a/replica:0/task:0/device:CPU:0"; variable