Searched refs:batchnorm_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 993 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormInferenceArgs() local
995 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormInferenceArgs()
1001 status = wrap::miopenSetOpArgsBatchNormInference(fusion_args_, batchnorm_op, in SetBatchNormInferenceArgs()
1018 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormForwardArgs() local
1020 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormForwardArgs()
1029 fusion_args_, batchnorm_op, alpha, beta, scale, offset, saved_mean, in SetBatchNormForwardArgs()
1045 miopenFusionOpDescriptor_t batchnorm_op; in SetBatchNormBackwardArgs() local
1047 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormBackwardArgs()
1054 fusion_args_, batchnorm_op, alpha, beta, x, scale, offset, scale_grad, in SetBatchNormBackwardArgs()
1247 miopenFusionOpDescriptor_t batchnorm_op; in ScopedFusionPlanBatchNormActivationInference() local
[all …]
|