Searched refs:actv_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1067 miopenFusionOpDescriptor_t actv_op; in SetActivationForwardArgs() local
1068 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationForwardArgs()
1075 wrap::miopenSetOpArgsActivForward(fusion_args_, actv_op, alpha, beta, in SetActivationForwardArgs()
1089 miopenFusionOpDescriptor_t actv_op; in SetActivationBackwardArgs() local
1090 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationBackwardArgs()
1096 status = wrap::miopenSetOpArgsActivBackward(fusion_args_, actv_op, alpha, in SetActivationBackwardArgs()
1148 miopenFusionOpDescriptor_t actv_op; in ScopedFusionPlanConvolutionBiasActivation() local
1150 fusion_plan_, &actv_op, in ScopedFusionPlanConvolutionBiasActivation()
1257 miopenFusionOpDescriptor_t actv_op; in ScopedFusionPlanBatchNormActivationInference() local
1259 fusion_plan_, &actv_op, in ScopedFusionPlanBatchNormActivationInference()
[all …]
|