Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h355 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
368 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
381 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Dcuda_dnn.cc567 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
2986 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
3042 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3180 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3759 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3781 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3803 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3826 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3857 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
4624 const dnn::FilterDescriptor& filter_descriptor, in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
392 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
405 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Drocm_dnn.cc710 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
1232 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1312 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue()
2846 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2942 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3100 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
3122 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3331 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3678 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3694 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.h329 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm()
437 const dnn::FilterDescriptor &filter_descriptor, in ConvolveBackwardDataWithAlgorithm()
474 const dnn::FilterDescriptor &filter_descriptor, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1146 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
1164 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
1183 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
1204 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
1245 const FilterDescriptor& filter_descriptor, in PrepareForConvolution()
1307 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
1408 Stream* stream, const dnn::FilterDescriptor& filter_descriptor, in DoConvolveBackwardData()
1460 const FilterDescriptor& filter_descriptor, in DoConvolveBackwardFilter()
2490 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolutionBiasActivation()
2682 const FilterDescriptor& filter_descriptor, DeviceMemoryBase filter_data, in DoPrepareForConvolution()
Dstream.cc449 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
480 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
511 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
543 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
574 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
605 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
624 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
651 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
678 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier, in ThenSeparableConvolve()
Dstream_executor_pimpl.cc280 const dnn::FilterDescriptor &filter_descriptor, in GetMIOpenConvolveAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h71 se::dnn::FilterDescriptor filter_descriptor; member
Dgpu_conv_runner.cc387 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig() local