| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.h | 538 class ConvolutionDescriptor {
543 ConvolutionDescriptor();
544 explicit ConvolutionDescriptor(int ndims);
545 ~ConvolutionDescriptor();
551 ConvolutionDescriptor& set_zero_padding_height(int64_t value) { in set_zero_padding_height()
555 ConvolutionDescriptor& set_zero_padding_width(int64_t value) { in set_zero_padding_width()
559 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64_t value) { in set_zero_padding()
563 ConvolutionDescriptor& set_vertical_filter_stride(int64_t value) { in set_vertical_filter_stride()
567 ConvolutionDescriptor& set_horizontal_filter_stride(int64_t value) { in set_horizontal_filter_stride()
571 ConvolutionDescriptor& set_filter_stride(DimIndex dim, int64_t value) { in set_filter_stride()
[all …]
|
| D | dnn.cc | 125 const dnn::ConvolutionDescriptor& /*convolution_descriptor*/, in GetConvolveRunners() argument
138 const dnn::ConvolutionDescriptor& convolution_descriptor) { in ConvolveRunnerFromDesc()
151 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
167 const dnn::ConvolutionDescriptor& convolution_descriptor, in FusedConvolveRunnerFromDesc()
181 const dnn::ConvolutionDescriptor& /*convolution_descriptor*/, in GetMIOpenConvolveAlgorithms() argument
738 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
746 ConvolutionDescriptor::ConvolutionDescriptor() in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
747 : ConvolutionDescriptor(/*ndims=*/2) {} in ConvolutionDescriptor()
749 ConvolutionDescriptor::~ConvolutionDescriptor() {} in ~ConvolutionDescriptor()
751 std::string ConvolutionDescriptor::ToString() const { in ToString()
[all …]
|
| D | lazy_op_runner.h | 126 const ConvolutionDescriptor& convolution_descriptor;
151 const ConvolutionDescriptor& convolution_descriptor;
|
| D | stream.h | 69 class ConvolutionDescriptor; variable
311 const dnn::ConvolutionDescriptor &convolution_descriptor,
321 const dnn::ConvolutionDescriptor &convolution_descriptor,
331 const dnn::ConvolutionDescriptor &convolution_descriptor,
343 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithAlgorithm()
372 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
399 const dnn::ConvolutionDescriptor &convolution_descriptor) { in ConvolveRunnerFromDesc()
419 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveRunnerFromDesc()
438 const dnn::ConvolutionDescriptor &convolution_descriptor,
|
| D | stream_executor_pimpl.h | 380 const dnn::ConvolutionDescriptor& convolution_descriptor,
393 const dnn::ConvolutionDescriptor& convolution_descriptor,
406 const dnn::ConvolutionDescriptor& convolution_descriptor,
|
| D | stream_executor_pimpl.cc | 300 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
323 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetFusedConvolveRunners()
343 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
|
| D | stream.cc | 47 std::string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) { in ToVlogString()
442 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
463 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
490 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
517 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenSeparableConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 248 const dnn::ConvolutionDescriptor& convolution_descriptor,
259 const dnn::ConvolutionDescriptor& convolution_descriptor) override;
268 const dnn::ConvolutionDescriptor& convolution_descriptor,
347 const dnn::ConvolutionDescriptor& convolution_descriptor,
358 const dnn::ConvolutionDescriptor& convolution_descriptor,
373 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
386 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
399 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
512 const dnn::ConvolutionDescriptor& convolution_descriptor,
527 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | rocm_dnn.cc | 69 using dnn::ConvolutionDescriptor;
812 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2909 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2959 ConvolutionDescriptor conv_descriptor) in RocmConvRunner()
3118 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3153 const dnn::ConvolutionDescriptor& convolution_descriptor, bool use_fallback, in GetConvolveRunners()
3192 const dnn::ConvolutionDescriptor& convolution_descriptor) { in ConvolveRunnerFromDesc()
3218 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3240 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3449 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 228 const dnn::ConvolutionDescriptor& convolution_descriptor,
239 const dnn::ConvolutionDescriptor& convolution_descriptor) override;
250 const dnn::ConvolutionDescriptor& convolution_descriptor,
265 const dnn::ConvolutionDescriptor& convolution_descriptor,
343 const dnn::ConvolutionDescriptor& convolution_descriptor,
354 const dnn::ConvolutionDescriptor& convolution_descriptor,
369 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
382 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
395 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
506 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | cuda_dnn.cc | 531 using ConvolutionDescriptor = typedef
564 ConvolutionDescriptor CreateConvolutionDescriptor() { in CreateConvolutionDescriptor()
567 return ConvolutionDescriptor(result); in CreateConvolutionDescriptor()
841 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
896 ConvolutionDescriptor handle_; // Owned.
3057 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionForwardAlgorithm()
3120 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardDataAlgorithm()
3181 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardFilterAlgorithm()
3509 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnOperationGraph()
3636 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnFusedOperationGraph()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.cc | 114 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneFusedConv()
229 const se::dnn::ConvolutionDescriptor& conv_desc,
245 const se::dnn::ConvolutionDescriptor& conv_desc,
262 const se::dnn::ConvolutionDescriptor& conv_desc,
279 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneUnfusedConv()
450 const se::dnn::ConvolutionDescriptor& conv_desc,
462 const se::dnn::ConvolutionDescriptor& conv_desc,
475 const se::dnn::ConvolutionDescriptor& conv_desc,
|
| D | conv_ops_gpu.h | 109 const se::dnn::ConvolutionDescriptor& conv_desc,
124 const se::dnn::ConvolutionDescriptor& conv_desc,
168 const se::dnn::ConvolutionDescriptor& conv_desc, in LaunchAutotunedConv()
|
| D | gpu_utils.cc | 145 const se::dnn::ConvolutionDescriptor& conv_desc, in LogConvAutotuneResults()
188 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults()
|
| D | gpu_utils.h | 264 const se::dnn::ConvolutionDescriptor& conv_desc,
276 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale,
|
| D | conv_grad_input_ops.cc | 263 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| D | conv_ops_3d.cc | 426 se::dnn::ConvolutionDescriptor conv_desc(3); in launch()
|
| D | conv_ops_fused_impl.h | 510 se::dnn::ConvolutionDescriptor conv_desc;
|
| /external/android-nn-driver/test/ |
| D | Dilation.hpp | 77 template<typename ConvolutionDescriptor>
78 void CheckDilationParams(const ConvolutionDescriptor& descriptor) in CheckDilationParams()
|
| /external/pytorch/aten/src/ATen/miopen/ |
| D | Descriptors.h | 109 struct ConvolutionDescriptor struct
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 64 se::dnn::ConvolutionDescriptor conv_desc;
|
| D | gpu_conv_runner.cc | 62 using se::dnn::ConvolutionDescriptor;
421 config.conv_desc = ConvolutionDescriptor(effective_num_dimensions); in GetGpuConvConfig()
|
| /external/pytorch/aten/src/ATen/cudnn/ |
| D | Descriptors.h | 196 struct TORCH_CUDA_CPP_API ConvolutionDescriptor struct
|
| /external/armnn/shim/sl/canonical/ |
| D | ConversionUtils.hpp | 873 template<typename ConvolutionDescriptor>
876 ConvolutionDescriptor& descriptor, in GetOptionalConvolutionDilationParams()
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | Conv_v7.cpp | 81 ConvolutionDescriptor cdesc;
|