Home
last modified time | relevance | path

Searched refs:ConvolutionDescriptor (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h508 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 …]
Ddnn.cc455 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()
Dstream.h69 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 …]
Dstream.cc46 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/
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()
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 …]
Dcuda_dnn.cc421 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/
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()
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 …]
Drocm_dnn.cc61 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/
Dcudnn_conv_runner.cc35 using se::dnn::ConvolutionDescriptor;
205 ConvolutionDescriptor convolution_descriptor(effective_num_dimensions); in RunCudnnConvImpl()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc371 se::dnn::ConvolutionDescriptor conv_desc(3); in launch()
Dconv_grad_ops_3d.cc1274 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
1673 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute()
Dconv_ops_fused_impl.h695 se::dnn::ConvolutionDescriptor conv_desc;
Dconv_ops.cc780 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
Dconv_grad_filter_ops.cc733 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
Dconv_grad_input_ops.cc849 se::dnn::ConvolutionDescriptor conv_desc; in operator ()()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc611 dnn::ConvolutionDescriptor conv_desc; in launch()