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.h280 const dnn::BatchDescriptor& bias_descriptor,
294 const dnn::BatchDescriptor& bias_descriptor,
310 const dnn::BatchDescriptor& bias_descriptor,
326 const dnn::BatchDescriptor& bias_descriptor,
375 const dnn::BatchDescriptor& bias_descriptor,
381 const dnn::BatchDescriptor& bias_descriptor,
387 const dnn::BatchDescriptor& bias_descriptor,
590 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor,
602 const dnn::BatchDescriptor& bias_descriptor,
Dcuda_dnn.cc2929 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl() argument
2952 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl()
3350 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3360 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3374 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3384 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3398 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3409 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3423 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3441 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h271 const dnn::BatchDescriptor& bias_descriptor,
285 const dnn::BatchDescriptor& bias_descriptor,
301 const dnn::BatchDescriptor& bias_descriptor,
317 const dnn::BatchDescriptor& bias_descriptor,
366 const dnn::BatchDescriptor& bias_descriptor,
372 const dnn::BatchDescriptor& bias_descriptor,
378 const dnn::BatchDescriptor& bias_descriptor,
539 const dnn::BatchDescriptor& bias_descriptor,
659 const dnn::BatchDescriptor& bias_descriptor,
713 const dnn::BatchDescriptor& bias_descriptor,
Drocm_dnn.cc1121 miopenTensorDescriptor_t bias_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1127 bias_descriptor, activation_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1142 bias_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1201 miopenTensorDescriptor_t bias_descriptor, in GetFusionOpHashValue() argument
1215 tensorflow::Hash64Combine(hash_value, GetHashValue(bias_descriptor)); in GetFusionOpHashValue()
3152 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3169 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3186 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3204 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolve() argument
3231 const dnn::BatchDescriptor& bias_descriptor, in DoConvolveBackwardBiasImpl() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1122 const dnn::BatchDescriptor& bias_descriptor,
1139 const dnn::BatchDescriptor& bias_descriptor,
1157 const dnn::BatchDescriptor& bias_descriptor,
1177 const dnn::BatchDescriptor& bias_descriptor,
1431 const BatchDescriptor& bias_descriptor,
1439 const BatchDescriptor& bias_descriptor,
1447 const BatchDescriptor& bias_descriptor,
2358 const dnn::BatchDescriptor& bias_descriptor,
Dstream.h344 const dnn::BatchDescriptor &bias_descriptor,
358 const dnn::BatchDescriptor &bias_descriptor,
372 const dnn::BatchDescriptor &bias_descriptor,
387 const dnn::BatchDescriptor &bias_descriptor,
478 const dnn::BatchDescriptor &bias_descriptor,
483 const dnn::BatchDescriptor &bias_descriptor,
489 const dnn::BatchDescriptor &bias_descriptor,
2029 const dnn::BatchDescriptor &bias_descriptor,
Dstream.cc447 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm() argument
465 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
485 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm() argument
503 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
523 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm() argument
534 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode), in ThenFusedConvolveWithAlgorithm()
542 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
562 const dnn::BatchDescriptor &bias_descriptor, in ThenFusedConvolveWithAlgorithm() argument
572 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode), in ThenFusedConvolveWithAlgorithm()
580 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
[all …]