Home
last modified time | relevance | path

Searched defs:conv_input_descriptor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3146 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3163 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3180 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3198 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
4023 const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl()
4111 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2923 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl()
3344 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3368 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3392 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
3417 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc441 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()
479 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()
517 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()
556 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()