Searched defs:conv_input_descriptor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3435 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3452 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3469 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3487 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 4311 const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl() 4399 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3214 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl() 3779 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3803 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3827 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3852 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() 3885 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 461 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() 499 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() 537 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() 576 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() 614 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()
|