| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 508 class ConvolutionDescriptor {
513 ConvolutionDescriptor();
514 explicit ConvolutionDescriptor(int ndims);
515 ~ConvolutionDescriptor();
521 ConvolutionDescriptor& set_zero_padding_height(int64 value) {
525 ConvolutionDescriptor& set_zero_padding_width(int64 value) {
529 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64 value) {
533 ConvolutionDescriptor& set_vertical_filter_stride(int64 value) {
537 ConvolutionDescriptor& set_horizontal_filter_stride(int64 value) {
541 ConvolutionDescriptor& set_filter_stride(DimIndex dim, int64 value) {
[all …]
|
| D | dnn.cc | 455 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
463 ConvolutionDescriptor::ConvolutionDescriptor() in ConvolutionDescriptor() function in stream_executor::dnn::ConvolutionDescriptor
464 : ConvolutionDescriptor(/*ndims=*/2) {} in ConvolutionDescriptor()
466 ConvolutionDescriptor::~ConvolutionDescriptor() {} in ~ConvolutionDescriptor()
468 string ConvolutionDescriptor::ToString() const { in ToString()
485 string ConvolutionDescriptor::ToShortString() const { in ToShortString()
|
| D | stream.h | 69 class ConvolutionDescriptor; variable
280 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,
[all …]
|
| D | stream.cc | 46 string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) { in ToVlogString()
445 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
483 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
521 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
560 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
598 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
640 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
682 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveWithAlgorithm()
724 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolve()
740 const dnn::ConvolutionDescriptor &convolution_descriptor, in ThenConvolveQuantized()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 269 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()
352 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
365 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
540 const dnn::ConvolutionDescriptor& convolution_descriptor,
588 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | cuda_dnn.cc | 421 using ConvolutionDescriptor = typedef
452 ConvolutionDescriptor CreateConvolutionDescriptor() { in CreateConvolutionDescriptor()
455 return ConvolutionDescriptor(result); in CreateConvolutionDescriptor()
650 const dnn::ConvolutionDescriptor& convolution_descriptor, in CudnnConvolutionDescriptor()
709 ConvolutionDescriptor handle_; // Owned.
2631 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2689 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
2927 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolveImpl()
3348 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3372 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 260 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()
343 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolveQuantized()
356 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoSeparableConvolve()
523 const dnn::ConvolutionDescriptor& convolution_descriptor,
538 const dnn::ConvolutionDescriptor& convolution_descriptor,
[all …]
|
| D | rocm_dnn.cc | 61 using dnn::ConvolutionDescriptor;
647 const ConvolutionDescriptor& convolution_descriptor, in ScopedConvolutionDescriptor()
2597 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoPrepareForConvolution()
2815 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoConvolve()
3150 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3167 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3184 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3202 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolve()
3989 const dnn::ConvolutionDescriptor& convolution_descriptor, in DeriveOutputBatchDescriptor()
4027 const dnn::ConvolutionDescriptor& convolution_descriptor, in DoFusedConvolutionBiasActivationImpl()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 35 using se::dnn::ConvolutionDescriptor;
205 ConvolutionDescriptor convolution_descriptor(effective_num_dimensions); in RunCudnnConvImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 371 se::dnn::ConvolutionDescriptor conv_desc(3); in launch()
|
| D | conv_grad_ops_3d.cc | 1274 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
1673 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
|
| D | conv_ops_fused_impl.h | 695 se::dnn::ConvolutionDescriptor conv_desc;
|
| D | conv_ops.cc | 780 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| D | conv_grad_filter_ops.cc | 733 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| D | conv_grad_input_ops.cc | 849 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 611 dnn::ConvolutionDescriptor conv_desc; in launch()
|