Home
last modified time | relevance | path

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 Ddepthwise_conv_op_d9m_test.py37 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 DLossCTC.cpp97 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 DLossCTC.cpp501 bool use_cudnn = in ctc_loss_impl() local
543 bool use_cudnn = in ctc_loss() local
H A DConvolution.cpp419 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 Ddepthwise_conv_grad_op.cc644 bool use_cudnn = in Compute() local
1149 bool use_cudnn = std::is_same<Device, GPUDevice>::value && in Compute() local
H A Dconv_grad_input_ops.cc74 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
H A Ddepthwise_conv_op.cc429 bool use_cudnn = in Compute() local
H A Dconv_grad_filter_ops.cc670 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
H A Dconv_ops.cc802 OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, in operator ()()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dctc_ops.py202 use_cudnn=False): argument