Searched refs:fusion_plan_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 916 fusion_plan_(nullptr), in ScopedFusionPlanBase()
939 miopen_handle_, fusion_plan_, input_descriptor, input_data, in Execute()
955 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &conv_op); in SetConvolutionArgs()
973 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &bias_op); in SetBiasArgs()
995 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormInferenceArgs()
1020 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormForwardArgs()
1047 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormBackwardArgs()
1068 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationForwardArgs()
1090 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationBackwardArgs()
1107 miopenFusionPlanDescriptor_t fusion_plan_; member in stream_executor::gpu::ScopedFusionPlanBase
[all …]
|