Searched refs:tuple_shape (Results 1 – 5 of 5) sorted by relevance
39 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in IsNodeOutputDynamicShape() local40 MS_EXCEPTION_IF_NULL(tuple_shape); in IsNodeOutputDynamicShape()41 for (size_t i = 0; i < tuple_shape->size(); i++) { in IsNodeOutputDynamicShape()42 auto b_shape = (*tuple_shape)[i]; in IsNodeOutputDynamicShape()
53 auto tuple_shape = cnode_shape->cast<abstract::TupleShapePtr>(); in GenNewTensor() local54 MS_EXCEPTION_IF_NULL(tuple_shape); in GenNewTensor()60 auto shape_elem = tuple_shape->shape()[i]; in GenNewTensor()
138 auto tuple_shape = tuple_shape_ptr->shape(); in GetNodeShape() local139 for (auto &shape : tuple_shape) { in GetNodeShape()
819 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetOutputInferShape() local820 MS_EXCEPTION_IF_NULL(tuple_shape); in GetOutputInferShape()821 if (output_idx >= tuple_shape->size()) { in GetOutputInferShape()822 …CEPTION) << "Output index " << output_idx << "is larger than output number " << tuple_shape->size() in GetOutputInferShape()826 auto b_shp = (*tuple_shape)[output_idx]; in GetOutputInferShape()1176 auto tuple_shape = base_shape->cast<abstract::TupleShapePtr>(); in GetOutputDetailShape() local1177 MS_EXCEPTION_IF_NULL(tuple_shape); in GetOutputDetailShape()1178 if (output_idx >= tuple_shape->size()) { in GetOutputDetailShape()1179 …CEPTION) << "Output index " << output_idx << "is larger than output number " << tuple_shape->size() in GetOutputDetailShape()1183 auto b_shp = (*tuple_shape)[output_idx]; in GetOutputDetailShape()[all …]
426 auto tuple_shape = shape->cast<abstract::TupleShapePtr>()->shape(); in SetShapeToNodeProto() local428 SetShapeToNodeProto(elements[i], tuple_shape[i], attr_proto, seq_string); in SetShapeToNodeProto()