Searched refs:scale_offset_mean_variance_descriptor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 1402 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference() argument
1407 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1417 scale_offset_mean_variance_descriptor); in ScopedFusionPlanBatchNormActivationInference()
1472 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1483 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1502 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward() argument
1507 scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1572 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() argument
1583 hash_value, GetHashValue(scale_offset_mean_variance_descriptor)); in GetFusionOpHashValue()
1603 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationBackward() argument
[all …]
|
| H A D | rocm_dnn.h | 559 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
570 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
583 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
599 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
726 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
737 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
753 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2464 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() argument
2477 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() argument
2518 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() argument
2531 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() argument
2578 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward() argument
2596 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward() argument
|