Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_fusion.cc82 auto fused_batch_norm_with_add_relu = graph->NewCNode(inputs); in Process() local
83 MS_EXCEPTION_IF_NULL(fused_batch_norm_with_add_relu); in Process()
92 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_with_add_relu.ge… in Process()
93 AnfAlgo::CopyNodeAttrs(batch_norm, fused_batch_norm_with_add_relu); in Process()
97 manager->Replace(batch_norm, fused_batch_norm_with_add_relu); in Process()
98 device::gpu::SetKernelInfo(fused_batch_norm_with_add_relu); in Process()
Dpost_batch_norm_add_relu_fusion.cc82 auto fused_batch_norm_with_add_relu = graph->NewCNode(inputs); in Process() local
83 MS_EXCEPTION_IF_NULL(fused_batch_norm_with_add_relu); in Process()
92 …AnfAlgo::SetOutputInferTypeAndShape(outputs_type, outputs_shape, fused_batch_norm_with_add_relu.ge… in Process()
93 AnfAlgo::CopyNodeAttrs(batch_norm, fused_batch_norm_with_add_relu); in Process()
97 manager->Replace(batch_norm, fused_batch_norm_with_add_relu); in Process()
98 device::gpu::SetKernelInfo(fused_batch_norm_with_add_relu); in Process()