Searched refs:fusion_plan (Results 1 – 1 of 1) sorted by relevance
318 miopenFusionPlanDescriptor_t* fusion_plan, in FindOrCreate() argument327 *fusion_plan = it->second; in FindOrCreate()330 auto status = wrap::miopenCreateFusionPlan(fusion_plan, fusion_direction, in FindOrCreate()336 cached_plans[hash] = *fusion_plan; in FindOrCreate()4051 ScopedFusionPlanConvolutionBiasActivation fusion_plan{ in DoFusedConvolutionBiasActivationImpl() local4057 if (fusion_plan.CompilationSucceeded()) { in DoFusedConvolutionBiasActivationImpl()4070 fusion_plan.SetConvolutionArgs(filter_data.opaque()); in DoFusedConvolutionBiasActivationImpl()4074 status = fusion_plan.SetBiasArgs(bias_data.opaque()); in DoFusedConvolutionBiasActivationImpl()4078 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedConvolutionBiasActivationImpl()4083 fusion_plan.Execute(conv_input_nd.handle(), conv_input_data.opaque(), in DoFusedConvolutionBiasActivationImpl()[all …]