Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1121 miopenTensorDescriptor_t bias_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1201 miopenTensorDescriptor_t bias_descriptor, in GetFusionOpHashValue()
3152 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3169 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3186 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3204 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3231 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBiasImpl()
3259 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
3268 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
3278 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2929 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl()
3350 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3374 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3398 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3423 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve()
3474 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBiasImpl()
3495 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
3506 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
3517 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias()
3673 CudnnTensorDescriptor bias_descriptor(bias_dimensions, CUDNN_DATA_FLOAT); in DoBiasAdd() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc447 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm()
485 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm()
523 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm()
562 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm()
1095 const dnn::BatchDescriptor &bias_descriptor, in ThenConvolveBackwardBiasImpl()
1115 const dnn::BatchDescriptor &bias_descriptor, in ThenConvolveBackwardBias()
1124 const dnn::BatchDescriptor &bias_descriptor, in ThenConvolveBackwardBias()
1133 const dnn::BatchDescriptor &bias_descriptor, in ThenConvolveBackwardBias()