Home
last modified time | relevance | path

Searched refs:scale_offset_mean_variance_descriptor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1235 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference() argument
1240 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1250 scale_offset_mean_variance_descriptor); in ScopedFusionPlanBatchNormActivationInference()
1305 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1316 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1335 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward() argument
1340 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1405 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1416 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1436 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationBackward() argument
[all …]
Drocm_dnn.h570 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
581 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
594 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
610 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
723 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
734 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
750 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2395 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
2408 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
2449 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
2462 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
2509 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
2527 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,