Searched refs:fusion_plan_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1033 fusion_plan_(nullptr), in ScopedFusionPlanBase()
1056 miopen_handle_, fusion_plan_, input_descriptor, input_data, in Execute()
1072 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &conv_op); in SetConvolutionArgs()
1090 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &bias_op); in SetBiasArgs()
1112 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormInferenceArgs()
1135 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormForwardArgs()
1160 wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &batchnorm_op); in SetBatchNormBackwardArgs()
1181 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationForwardArgs()
1203 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &actv_op); in SetActivationBackwardArgs()
1220 miopenFusionPlanDescriptor_t fusion_plan_; member in stream_executor::gpu::ScopedFusionPlanBase
[all …]
|