| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 500 class ConvolutionDescriptor {
505 ConvolutionDescriptor();
506 explicit ConvolutionDescriptor(int ndims);
507 ~ConvolutionDescriptor();
513 ConvolutionDescriptor& set_zero_padding_height(int64 value) { in set_zero_padding_height()
517 ConvolutionDescriptor& set_zero_padding_width(int64 value) { in set_zero_padding_width()
521 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64 value) { in set_zero_padding()
525 ConvolutionDescriptor& set_vertical_filter_stride(int64 value) { in set_vertical_filter_stride()
529 ConvolutionDescriptor& set_horizontal_filter_stride(int64 value) { in set_horizontal_filter_stride()
533 ConvolutionDescriptor& set_filter_stride(DimIndex dim, int64 value) { in set_filter_stride()
[all …]
|
| D | dnn.cc | 58 const dnn::ConvolutionDescriptor& /*convolution_descriptor*/, in GetMIOpenConvolveAlgorithms() argument
486 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
494 ConvolutionDescriptor::ConvolutionDescriptor() in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
495 : ConvolutionDescriptor(/*ndims=*/2) {} in ConvolutionDescriptor()
497 ConvolutionDescriptor::~ConvolutionDescriptor() {} in ~ConvolutionDescriptor()
499 std::string ConvolutionDescriptor::ToString() const { in ToString()
515 std::string ConvolutionDescriptor::ToShortString() const { in ToShortString()
|
| D | stream.h | 70 class ConvolutionDescriptor; variable
301 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()
359 const dnn::ConvolutionDescriptor &convolution_descriptor,
373 const dnn::ConvolutionDescriptor &convolution_descriptor,
387 const dnn::ConvolutionDescriptor &convolution_descriptor,
402 const dnn::ConvolutionDescriptor &convolution_descriptor,
416 const dnn::ConvolutionDescriptor &convolution_descriptor,
[all …]
|
| D | stream_executor_pimpl.h | 377 const dnn::ConvolutionDescriptor &convolution_descriptor,
|
| D | stream.cc | 45 std::string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) { in ToVlogString()
451 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
482 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
513 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
545 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
576 const dnn::ConvolutionDescriptor &convolution_descriptor, in FusedConvolveWithAlgorithm()
607 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
627 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
654 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
681 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenSeparableConvolve()
|
| D | stream_executor_pimpl.cc | 283 const dnn::ConvolutionDescriptor &convolution_descriptor, in GetMIOpenConvolveAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 241 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()
395 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
408 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
575 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | rocm_dnn.cc | 65 using dnn::ConvolutionDescriptor;
758 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2849 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2945 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3103 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithms()
3125 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3334 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3680 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3696 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3712 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 276 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()
371 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
384 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
559 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | cuda_dnn.cc | 434 using ConvolutionDescriptor = typedef
467 ConvolutionDescriptor CreateConvolutionDescriptor() { in CreateConvolutionDescriptor()
470 return ConvolutionDescriptor(result); in CreateConvolutionDescriptor()
683 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
738 ConvolutionDescriptor handle_; // Owned.
2692 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionForwardAlgorithm()
2755 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardDataAlgorithm()
2816 const dnn::ConvolutionDescriptor& convolution_descriptor, in GetCudnnConvolutionBackwardFilterAlgorithm()
2989 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
3045 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 225 const se::dnn::ConvolutionDescriptor& conv_desc,
237 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale,
|
| D | gpu_utils.cc | 134 const se::dnn::ConvolutionDescriptor& conv_desc, in LogConvAutotuneResults()
177 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults()
|
| D | conv_grad_input_ops.cc | 267 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| D | conv_ops_3d.cc | 418 se::dnn::ConvolutionDescriptor conv_desc(3); in launch()
|
| D | conv_ops_fused_impl.h | 589 se::dnn::ConvolutionDescriptor conv_desc;
|
| D | conv_grad_ops_3d.cc | 1319 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
1816 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
|
| D | conv_grad_filter_ops.cc | 872 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| D | conv_ops.cc | 897 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 73 se::dnn::ConvolutionDescriptor conv_desc;
|
| D | gpu_conv_runner.cc | 36 using se::dnn::ConvolutionDescriptor;
400 config.conv_desc = ConvolutionDescriptor(effective_num_dimensions); in GetGpuConvConfig()
|