Home
last modified time | relevance | path

Searched defs:filter_descriptor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h336 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
349 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
362 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Dcuda_dnn.cc545 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
2628 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2686 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
2925 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3346 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3370 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3394 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3419 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
4091 const dnn::FilterDescriptor& filter_descriptor, in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h327 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
340 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
353 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Drocm_dnn.cc594 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
1119 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1199 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue()
2594 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2812 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3148 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3165 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3182 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3200 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3988 const FilterDescriptor& filter_descriptor, in DeriveOutputBatchDescriptor()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc193 FilterDescriptor filter_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc443 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
481 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
519 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
558 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
596 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
638 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
680 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
722 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
737 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
767 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
[all …]