Searched refs:gamma_shape_ptr (Results 1 – 3 of 3) sorted by relevance
37 auto gamma_shape_ptr = std::make_shared<abstract::Shape>(gamma_shape); in LayerNormBetaGammaBackpropInferShape() local38 …hared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{gamma_shape_ptr, gamma_shape_ptr}); in LayerNormBetaGammaBackpropInferShape()
33 auto gamma_shape_ptr = std::make_shared<abstract::Shape>(gamma_shape); in LayerNormBetaGammaBackpropV2InferShape() local34 …hared<abstract::TupleShape>(std::vector<abstract::BaseShapePtr>{gamma_shape_ptr, gamma_shape_ptr}); in LayerNormBetaGammaBackpropV2InferShape()
179 ShapePtr gamma_shape_ptr = std::make_shared<Shape>(gamma_shape, gamma_min_shape, gamma_max_shape); in InferImplBatchNorm() local180 …TensorPtr output_gamma = std::make_shared<AbstractTensor>(input_gamma->element(), gamma_shape_ptr); in InferImplBatchNorm()