Searched refs:lengths_shape (Results 1 – 2 of 2) sorted by relevance
908 ShapeVector lengths_shape = lengths->shape()->shape(); in InferImplSequenceMask() local911 lengths_shape_min = lengths_shape; in InferImplSequenceMask()915 lengths_shape_max = lengths_shape; in InferImplSequenceMask()918 lengths_shape.push_back(maxlen_value); in InferImplSequenceMask()922 …ShapePtr output_shape = std::make_shared<Shape>(lengths_shape, lengths_shape_min, lengths_shape_ma… in InferImplSequenceMask()
737 def check_shape(self, lengths_shape, maxlen_shape): argument738 validator.check("lengths_shape", len(lengths_shape), "", 0, Rel.GT, self.name)