Searched refs:fusion_args_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 917 fusion_args_(nullptr), in ScopedFusionPlanBase()
919 auto status = wrap::miopenCreateOperatorArgs(&fusion_args_); in ScopedFusionPlanBase()
927 auto status = wrap::miopenDestroyOperatorArgs(fusion_args_); in ~ScopedFusionPlanBase()
940 output_descriptor, output_data, fusion_args_); in Execute()
961 status = wrap::miopenSetOpArgsConvForward(fusion_args_, conv_op, alpha, in SetConvolutionArgs()
979 status = wrap::miopenSetOpArgsBiasForward(fusion_args_, bias_op, alpha, in SetBiasArgs()
1001 status = wrap::miopenSetOpArgsBatchNormInference(fusion_args_, batchnorm_op, in SetBatchNormInferenceArgs()
1029 fusion_args_, batchnorm_op, alpha, beta, scale, offset, saved_mean, in SetBatchNormForwardArgs()
1054 fusion_args_, batchnorm_op, alpha, beta, x, scale, offset, scale_grad, in SetBatchNormBackwardArgs()
1075 wrap::miopenSetOpArgsActivForward(fusion_args_, actv_op, alpha, beta, in SetActivationForwardArgs()
[all …]
|