Searched defs:conv_input_descriptor (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1144 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 1162 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 1181 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 1202 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 2488 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation()
|
D | stream.cc | 447 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() 478 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() 509 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() 541 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() 572 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 3178 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl() 3757 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3779 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3801 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3824 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3855 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 3676 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3692 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3708 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3725 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 4619 const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl() 4707 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation()
|