Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h361 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
374 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
387 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Dcuda_dnn.cc563 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
2896 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2955 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3216 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3781 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3805 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3829 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3854 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3887 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
4633 const dnn::FilterDescriptor& filter_descriptor, in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h346 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
359 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
372 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Drocm_dnn.cc643 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
1163 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1243 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue()
2639 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2805 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
2934 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
3437 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3454 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3471 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3489 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h63 se::dnn::FilterDescriptor filter_descriptor; member
Dgpu_conv_runner.cc391 FilterDescriptor& filter_descriptor = params.filter_descriptor; in GetGpuConvParams() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc463 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
501 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
539 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
578 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
616 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
654 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
696 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
738 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
780 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
822 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
[all …]
Dstream_executor_pimpl.cc296 const dnn::FilterDescriptor &filter_descriptor, in GetMIOpenConvolveAlgorithms()