Home
last modified time | relevance | path

Searched refs:fusion_plan (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc433 miopenFusionPlanDescriptor_t* fusion_plan, in FindOrCreate() argument
442 *fusion_plan = it->second; in FindOrCreate()
445 auto status = wrap::miopenCreateFusionPlan(fusion_plan, fusion_direction, in FindOrCreate()
451 cached_plans[hash] = *fusion_plan; in FindOrCreate()
4647 ScopedFusionPlanConvolutionBiasActivation fusion_plan{ in DoFusedConvolutionBiasActivationImpl() local
4653 if (fusion_plan.CompilationSucceeded()) { in DoFusedConvolutionBiasActivationImpl()
4666 fusion_plan.SetConvolutionArgs(filter_data.opaque()); in DoFusedConvolutionBiasActivationImpl()
4670 status = fusion_plan.SetBiasArgs(bias_data.opaque()); in DoFusedConvolutionBiasActivationImpl()
4674 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedConvolutionBiasActivationImpl()
4679 fusion_plan.Execute(conv_input_nd.handle(), conv_input_data.opaque(), in DoFusedConvolutionBiasActivationImpl()
[all …]