Home
last modified time | relevance | path

Searched defs:conv_desc (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/protobuf/
Dconv_autotuning.proto18 stream_executor.dnn.ConvolutionDescriptorProto conv_desc = 5; field
/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc145 const se::dnn::ConvolutionDescriptor& conv_desc, in LogConvAutotuneResults()
188 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults()
Dconv_ops_gpu.cc114 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneFusedConv()
279 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneUnfusedConv()
Dconv_ops_gpu.h168 const se::dnn::ConvolutionDescriptor& conv_desc, in LaunchAutotunedConv()
Dconv_grad_input_ops.cc263 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
Dconv_ops_3d.cc426 se::dnn::ConvolutionDescriptor conv_desc(3); in launch() local
Dconv_grad_ops_3d.cc1421 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1820 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
Dconv_grad_filter_ops.cc882 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
Dconv_ops.cc1048 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h64 se::dnn::ConvolutionDescriptor conv_desc; member
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc3569 auto conv_desc = in GetCudnnOperationGraph() local
3744 auto conv_desc = in GetCudnnFusedOperationGraph() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc401 uint64_t GetHashValue(miopenConvolutionDescriptor_t conv_desc) { in GetHashValue()