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 …]
|