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()
|