Home
last modified time | relevance | path

Searched refs:convolution_descriptor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h241 const dnn::ConvolutionDescriptor& convolution_descriptor,
314 const dnn::ConvolutionDescriptor& convolution_descriptor,
323 const dnn::ConvolutionDescriptor& convolution_descriptor,
337 const dnn::ConvolutionDescriptor& convolution_descriptor,
351 const dnn::ConvolutionDescriptor& convolution_descriptor,
367 const dnn::ConvolutionDescriptor& convolution_descriptor,
382 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
395 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
408 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve() argument
575 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Drocm_dnn.cc758 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor() argument
766 const auto& strides64 = convolution_descriptor.strides(); in ScopedConvolutionDescriptor()
767 const auto& padding64 = convolution_descriptor.padding(); in ScopedConvolutionDescriptor()
768 if (convolution_descriptor.pad_alignment() == in ScopedConvolutionDescriptor()
774 std::vector<int> strides(convolution_descriptor.ndims()); in ScopedConvolutionDescriptor()
775 std::vector<int> padding(convolution_descriptor.ndims()); in ScopedConvolutionDescriptor()
781 std::vector<int> upscale(convolution_descriptor.ndims()); in ScopedConvolutionDescriptor()
782 const auto& dilations64 = convolution_descriptor.dilations(); in ScopedConvolutionDescriptor()
787 handle_, convolution_descriptor.ndims(), padding.data(), strides.data(), in ScopedConvolutionDescriptor()
795 << convolution_descriptor.group_count(); in ScopedConvolutionDescriptor()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h276 const dnn::ConvolutionDescriptor& convolution_descriptor,
285 const dnn::ConvolutionDescriptor& convolution_descriptor,
299 const dnn::ConvolutionDescriptor& convolution_descriptor,
313 const dnn::ConvolutionDescriptor& convolution_descriptor,
329 const dnn::ConvolutionDescriptor& convolution_descriptor,
343 const dnn::ConvolutionDescriptor& convolution_descriptor,
358 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
371 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
384 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve() argument
559 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Dcuda_dnn.cc683 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor() argument
686 absl::Span<const int64> strides64 = convolution_descriptor.strides(); in CudnnConvolutionDescriptor()
687 absl::Span<const int64> padding64 = convolution_descriptor.padding(); in CudnnConvolutionDescriptor()
688 absl::Span<const int64> dilations64 = convolution_descriptor.dilations(); in CudnnConvolutionDescriptor()
689 CHECK_NE(convolution_descriptor.pad_alignment(), in CudnnConvolutionDescriptor()
694 std::vector<int> strides(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
695 std::vector<int> padding(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
696 std::vector<int> dilations(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
707 handle_.get(), convolution_descriptor.ndims(), padding.data(), in CudnnConvolutionDescriptor()
709 convolution_descriptor.convolution_not_crosscorr() in CudnnConvolutionDescriptor()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1148 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve() argument
1166 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve() argument
1185 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve() argument
1206 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve() argument
1249 const ConvolutionDescriptor& convolution_descriptor, in PrepareForConvolution() argument
1256 output_data, convolution_descriptor, algorithm_config, in PrepareForConvolution()
1300 const ConvolutionDescriptor& convolution_descriptor,
1309 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve() argument
1319 output_descriptor, *output_data, convolution_descriptor, in DoConvolve()
1337 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Dstream.h301 const dnn::ConvolutionDescriptor &convolution_descriptor,
311 const dnn::ConvolutionDescriptor &convolution_descriptor,
321 const dnn::ConvolutionDescriptor &convolution_descriptor,
331 const dnn::ConvolutionDescriptor &convolution_descriptor, in ConvolveWithAlgorithm() argument
342 convolution_descriptor, algorithm_config, scratch_allocator, in ConvolveWithAlgorithm()
348 *output, convolution_descriptor, algorithm_desc, scratch_memory, in ConvolveWithAlgorithm()
359 const dnn::ConvolutionDescriptor &convolution_descriptor,
373 const dnn::ConvolutionDescriptor &convolution_descriptor,
387 const dnn::ConvolutionDescriptor &convolution_descriptor,
402 const dnn::ConvolutionDescriptor &convolution_descriptor,
[all …]
Dstream.cc451 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm() argument
461 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in FusedConvolveWithAlgorithm()
469 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
482 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm() argument
492 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in FusedConvolveWithAlgorithm()
500 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
513 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm() argument
524 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in FusedConvolveWithAlgorithm()
532 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
545 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream_executor_pimpl.cc283 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetMIOpenConvolveAlgorithms() argument
293 convolution_descriptor, scratch_allocator, out_algorithms); in GetMIOpenConvolveAlgorithms()
Dstream_executor_pimpl.h377 const dnn::ConvolutionDescriptor &convolution_descriptor,