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.cc1348 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference() argument
1353 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1363 scale_offset_mean_variance_descriptor); in ScopedFusionPlanBatchNormActivationInference()
1418 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1429 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1448 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward() argument
1453 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1518 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1529 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1549 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationBackward() argument
[all …]
Drocm_dnn.h622 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
633 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
646 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
662 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
798 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
809 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
825 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2530 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() argument
2543 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() argument
2584 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() argument
2597 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() argument
2644 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward() argument
2662 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward() argument