Searched defs:conv_input_descriptor (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 3146 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/ |
D | cuda_dnn.cc | 2923 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/ |
D | stream.cc | 441 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()
|