| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_merge_variables_with_execute.cc | 154 bool check_device, bool check_same_region) { in BuildVariableAccessInfo()
459 bool check_device, bool check_same_region, OpBuilder* builder) { in MergeForOneTPUExecute()
|
| /aosp_15_r20/external/linux-kselftest/tools/testing/selftests/android/ion/ |
| H A D | ion_test.sh | 35 check_device() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDAUtils.h | 9 inline bool check_device(ArrayRef<Tensor> ts) { in check_device() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/xpu/ |
| H A D | XPUContext.cpp | 52 inline void check_device(DeviceIndex device) { in check_device() function
|
| H A D | XPUGeneratorImpl.cpp | 27 inline void check_device(DeviceIndex device) { in check_device() function
|
| /aosp_15_r20/external/ltp/libs/libltpuinput/ |
| H A D | tst_uinput.c | 96 static int check_device(void) in check_device() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | decompose_mem_bound_mm.py | 32 def check_device(a: Tensor, b: Tensor) -> bool: function
|
| H A D | pad_mm.py | 76 def check_device(a: Tensor, b: Tensor) -> bool: function
|
| /aosp_15_r20/external/gsc-utils/util/ |
| H A D | usb_if.c | 67 static libusb_device_handle *check_device(libusb_device *dev, uint16_t vid, in check_device() function
|
| /aosp_15_r20/external/pytorch/c10/xpu/ |
| H A D | XPUFunctions.cpp | 111 inline void check_device(DeviceIndex device) { in check_device() function
|
| H A D | XPUStream.cpp | 150 inline void check_device(DeviceIndex device) { in check_device() function
|
| /aosp_15_r20/external/ltp/testcases/kernel/input/ |
| H A D | input_helper.c | 296 static int check_device(void) in check_device() function
|
| /aosp_15_r20/external/igt-gpu-tools/overlay/kms/ |
| H A D | kms-overlay.c | 162 static int check_device(int fd) in check_device() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_v2.py | 1066 def check_device(path, device_string): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupUCC.cpp | 187 void check_device(c10::Device dev1, c10::Device dev2) { in check_device() function
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | _triton_ops.py | 33 def check_device(f_name, t, device): function
|
| /aosp_15_r20/external/linux-kselftest/tools/testing/selftests/net/ |
| H A D | nettest.c | 390 static int check_device(int sd, struct sock_args *args) in check_device() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 2620 def check_device(x, device_id): function
|