Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc205 ConvolutionDescriptor convolution_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
206 convolution_descriptor.set_group_count(feature_group_count); in RunCudnnConvImpl()
207 convolution_descriptor.set_convolution_not_crosscorr(dims_reversed); in RunCudnnConvImpl()
209 convolution_descriptor in RunCudnnConvImpl()
234 convolution_descriptor.set_zero_padding(static_cast<DimIndex>(0), 0) in RunCudnnConvImpl()
247 convolution_descriptor, output_descriptor, &output_buf, in RunCudnnConvImpl()
258 convolution_descriptor, input_descriptor, &input_buf, in RunCudnnConvImpl()
269 convolution_descriptor, filter_descriptor, &filter_buf, in RunCudnnConvImpl()
300 filter_descriptor, filter_buf, convolution_descriptor, side_input, in RunCudnnConvImpl()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h269 const dnn::ConvolutionDescriptor& convolution_descriptor,
278 const dnn::ConvolutionDescriptor& convolution_descriptor,
292 const dnn::ConvolutionDescriptor& convolution_descriptor,
307 const dnn::ConvolutionDescriptor& convolution_descriptor,
324 const dnn::ConvolutionDescriptor& convolution_descriptor,
339 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
352 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
365 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve() argument
540 const dnn::ConvolutionDescriptor& convolution_descriptor,
588 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Dcuda_dnn.cc650 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor() argument
653 absl::Span<const int64> strides64 = convolution_descriptor.strides(); in CudnnConvolutionDescriptor()
654 absl::Span<const int64> padding64 = convolution_descriptor.padding(); in CudnnConvolutionDescriptor()
655 absl::Span<const int64> dilations64 = convolution_descriptor.dilations(); in CudnnConvolutionDescriptor()
656 CHECK_NE(convolution_descriptor.pad_alignment(), in CudnnConvolutionDescriptor()
661 std::vector<int> strides(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
662 std::vector<int> padding(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
663 std::vector<int> dilations(convolution_descriptor.ndims()); in CudnnConvolutionDescriptor()
674 handle_.get(), convolution_descriptor.ndims(), padding.data(), in CudnnConvolutionDescriptor()
676 convolution_descriptor.convolution_not_crosscorr() in CudnnConvolutionDescriptor()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1120 const dnn::ConvolutionDescriptor& convolution_descriptor,
1137 const dnn::ConvolutionDescriptor& convolution_descriptor,
1155 const dnn::ConvolutionDescriptor& convolution_descriptor,
1175 const dnn::ConvolutionDescriptor& convolution_descriptor,
1195 const ConvolutionDescriptor& convolution_descriptor,
1202 output_data, convolution_descriptor, algorithm_config,
1245 const ConvolutionDescriptor& convolution_descriptor,
1254 const dnn::ConvolutionDescriptor& convolution_descriptor,
1264 convolution_descriptor, algorithm_desc, *scratch_memory,
1289 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Dstream.cc445 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm() argument
455 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in ThenFusedConvolveWithAlgorithm()
464 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
483 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm() argument
493 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in ThenFusedConvolveWithAlgorithm()
502 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
521 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm() argument
532 PARAM(filter_data), PARAM(convolution_descriptor), PARAM(biases), in ThenFusedConvolveWithAlgorithm()
541 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
560 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dstream.h280 const dnn::ConvolutionDescriptor &convolution_descriptor,
290 const dnn::ConvolutionDescriptor &convolution_descriptor,
300 const dnn::ConvolutionDescriptor &convolution_descriptor,
309 const dnn::ConvolutionDescriptor &convolution_descriptor,
320 const dnn::ConvolutionDescriptor &convolution_descriptor,
331 const dnn::ConvolutionDescriptor &convolution_descriptor,
342 const dnn::ConvolutionDescriptor &convolution_descriptor,
356 const dnn::ConvolutionDescriptor &convolution_descriptor,
370 const dnn::ConvolutionDescriptor &convolution_descriptor,
385 const dnn::ConvolutionDescriptor &convolution_descriptor,
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h260 const dnn::ConvolutionDescriptor& convolution_descriptor,
269 const dnn::ConvolutionDescriptor& convolution_descriptor,
283 const dnn::ConvolutionDescriptor& convolution_descriptor,
298 const dnn::ConvolutionDescriptor& convolution_descriptor,
315 const dnn::ConvolutionDescriptor& convolution_descriptor,
330 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
343 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized() argument
356 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve() argument
523 const dnn::ConvolutionDescriptor& convolution_descriptor,
538 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
Drocm_dnn.cc647 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor() argument
655 const auto& strides64 = convolution_descriptor.strides(); in ScopedConvolutionDescriptor()
656 const auto& padding64 = convolution_descriptor.padding(); in ScopedConvolutionDescriptor()
657 if (convolution_descriptor.pad_alignment() == in ScopedConvolutionDescriptor()
663 std::vector<int> strides(convolution_descriptor.ndims()); in ScopedConvolutionDescriptor()
664 std::vector<int> padding(convolution_descriptor.ndims()); in ScopedConvolutionDescriptor()
669 std::vector<int> upscale(convolution_descriptor.ndims(), 1); in ScopedConvolutionDescriptor()
680 << convolution_descriptor.group_count(); in ScopedConvolutionDescriptor()
682 handle_, convolution_descriptor.group_count()); in ScopedConvolutionDescriptor()
2597 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution() argument
[all …]