Home
last modified time | relevance | path

Searched defs:batchnorm_op (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1164 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormInferenceArgs() local
1187 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormForwardArgs() local
1212 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormBackwardArgs() local
1414 miopenFusionOpDescriptor_t batchnorm_op; in ScopedFusionPlanBatchNormActivationInference() local
1514 miopenFusionOpDescriptor_t batchnorm_op; in ScopedFusionPlanBatchNormActivationForward() local
1615 miopenFusionOpDescriptor_t batchnorm_op; in ScopedFusionPlanBatchNormActivationBackward() local