Home
last modified time | relevance | path

Searched refs:bias_descriptor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h287 const dnn::BatchDescriptor& bias_descriptor,
301 const dnn::BatchDescriptor& bias_descriptor,
315 const dnn::BatchDescriptor& bias_descriptor,
331 const dnn::BatchDescriptor& bias_descriptor,
345 const dnn::BatchDescriptor& bias_descriptor,
394 const dnn::BatchDescriptor& bias_descriptor,
400 const dnn::BatchDescriptor& bias_descriptor,
406 const dnn::BatchDescriptor& bias_descriptor,
624 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor,
636 const dnn::BatchDescriptor& bias_descriptor,
Dcuda_dnn.cc3184 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl() argument
3207 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl()
3763 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3772 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
3785 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3794 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
3807 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3817 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
3830 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3848 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h325 const dnn::BatchDescriptor& bias_descriptor,
339 const dnn::BatchDescriptor& bias_descriptor,
353 const dnn::BatchDescriptor& bias_descriptor,
369 const dnn::BatchDescriptor& bias_descriptor,
418 const dnn::BatchDescriptor& bias_descriptor,
424 const dnn::BatchDescriptor& bias_descriptor,
430 const dnn::BatchDescriptor& bias_descriptor,
591 const dnn::BatchDescriptor& bias_descriptor,
734 const dnn::BatchDescriptor& bias_descriptor,
788 const dnn::BatchDescriptor& bias_descriptor,
Drocm_dnn.cc1234 miopenTensorDescriptor_t bias_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1240 bias_descriptor, activation_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1255 bias_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1314 miopenTensorDescriptor_t bias_descriptor, in GetFusionOpHashValue() argument
1328 tensorflow::Hash64Combine(hash_value, GetHashValue(bias_descriptor)); in GetFusionOpHashValue()
3682 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3698 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3714 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3731 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3757 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBiasImpl() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1150 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
1168 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
1187 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
1208 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
1497 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias() argument
1505 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias() argument
1513 const BatchDescriptor& bias_descriptor, in DoConvolveBackwardBias() argument
2493 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolutionBiasActivation() argument
Dstream.h361 const dnn::BatchDescriptor &bias_descriptor,
375 const dnn::BatchDescriptor &bias_descriptor,
389 const dnn::BatchDescriptor &bias_descriptor,
404 const dnn::BatchDescriptor &bias_descriptor,
418 const dnn::BatchDescriptor &bias_descriptor,
501 const dnn::BatchDescriptor &bias_descriptor,
506 const dnn::BatchDescriptor &bias_descriptor,
512 const dnn::BatchDescriptor &bias_descriptor,
2096 const dnn::BatchDescriptor &bias_descriptor,
Dstream.cc453 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveWithAlgorithm() argument
470 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
484 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveWithAlgorithm() argument
501 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
515 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveWithAlgorithm() argument
526 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode), in FusedConvolveWithAlgorithm()
533 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
547 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveWithAlgorithm() argument
557 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode), in FusedConvolveWithAlgorithm()
564 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
[all …]