Searched refs:checkBackend (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/test/cpp_extensions/ |
| D | cudnn_extension.cpp | 33 torch::checkBackend(cudnn_relu_name, arg_inputs.tensor, torch::Backend::CUDA); in cudnn_relu_check()
36 torch::checkBackend( in cudnn_relu_check()
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | LossCTC.cpp | 181 checkBackend(c, {*log_probs}, Backend::CUDA); in _cudnn_ctc_loss()
182 checkBackend(c, {*targets}, Backend::CPU); in _cudnn_ctc_loss()
269 checkBackend(c, {*log_probs}, Backend::CUDA); in _cudnn_ctc_loss_tensor()
270 checkBackend(c, {*targets}, Backend::CUDA); in _cudnn_ctc_loss_tensor()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| D | Blas.cpp | 16 checkBackend("addmm_out", {result, self, mat1, mat2}, Backend::XPU); in addmm_out()
124 checkBackend("mm_out", {result, self, mat2}, Backend::XPU); in mm_out()
179 checkBackend("baddbmm_out", {input, batch1, batch2}, Backend::XPU); in baddbmm_out()
263 checkBackend("addbmm_out", {out, self, batch1, batch2}, Backend::XPU); in addbmm_out()
320 checkBackend("bmm_out", {result, self, batch2}, Backend::XPU); in bmm_out()
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorUtils.cpp | 233 static void checkBackend(CheckedFrom c, const Tensor& t, Backend backend) { in checkBackend() function
241 void checkBackend(CheckedFrom c, at::ArrayRef<Tensor> tensors, at::Backend backend) { in checkBackend() function
243 checkBackend(c, t, backend); in checkBackend()
|
| D | TensorUtils.h | 144 TORCH_API void checkBackend(
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NaiveDilatedConvolution.cpp | 151 checkBackend("slow_conv_dilated_location_check", {input, weight}, Backend::CPU); in slow_conv_dilated_location_check()
153 checkBackend("slow_conv_dilated_location_check", {bias}, Backend::CPU); in slow_conv_dilated_location_check()
156 checkBackend("slow_conv_dilated_location_check", {grad_output}, Backend::CPU); in slow_conv_dilated_location_check()
|
| D | Normalization.cpp | 770 checkBackend("batch_norm_cpu_out", {self, weight, bias, running_mean, running_var}, Backend::CPU); in batch_norm_cpu_out()
813 checkBackend("batch_norm_cpu", {self, weight, bias, running_mean, running_var}, Backend::CPU); in batch_norm_cpu()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | SummaryOps.cu | 183 checkBackend("CUDA_tensor_histogram", {a, b}, Backend::CUDA); in CUDA_tensor_histogram()
185 checkBackend("CUDA_tensor_histogram", {c}, Backend::CUDA); in CUDA_tensor_histogram()
|