/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()
|