Searched defs:use_cudnn (Results 1 – 10 of 10) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
H A D | depthwise_conv_op_d9m_test.py | 37 use_cudnn=False, argument 80 use_cudnn=False, argument 111 use_cudnn=False, argument
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
H A D | LossCTC.cpp | 97 bool use_cudnn = ctx.userEnabledCuDNN() && (BLANK == 0) && in _use_cudnn_ctc_loss() local 128 bool use_cudnn = ctx.userEnabledCuDNN() && (BLANK == 0) && in _use_cudnn_ctc_loss_tensor() local
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
H A D | LossCTC.cpp | 501 bool use_cudnn = in ctc_loss_impl() local 543 bool use_cudnn = in ctc_loss() local
|
H A D | Convolution.cpp | 419 bool use_cudnn(const at::Tensor& input, const at::Tensor& weight) const { in use_cudnn() function
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | depthwise_conv_grad_op.cc | 644 bool use_cudnn = in Compute() local 1149 bool use_cudnn = std::is_same<Device, GPUDevice>::value && in Compute() local
|
H A D | conv_grad_input_ops.cc | 74 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
H A D | depthwise_conv_op.cc | 429 bool use_cudnn = in Compute() local
|
H A D | conv_grad_filter_ops.cc | 670 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
H A D | conv_ops.cc | 802 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | ctc_ops.py | 202 use_cudnn=False): argument
|