Searched refs:gamma_max_shape (Results 1 – 1 of 1) sorted by relevance
175 ShapeVector gamma_max_shape = input_gamma->shape()->max_shape(); in InferImplBatchNorm() local176 CheckMinMaxShape(gamma_shape, &gamma_min_shape, &gamma_max_shape); in InferImplBatchNorm()179 ShapePtr gamma_shape_ptr = std::make_shared<Shape>(gamma_shape, gamma_min_shape, gamma_max_shape); in InferImplBatchNorm()