Searched refs:b_shp (Results 1 – 2 of 2) sorted by relevance
826 auto b_shp = (*tuple_shape)[output_idx]; in GetOutputInferShape() local827 if (b_shp->isa<abstract::Shape>()) { in GetOutputInferShape()828 return TransShapeToSizet(b_shp->cast<abstract::ShapePtr>()); in GetOutputInferShape()829 } else if (b_shp->isa<abstract::NoShape>()) { in GetOutputInferShape()1183 auto b_shp = (*tuple_shape)[output_idx]; in GetOutputDetailShape() local1184 if (b_shp->isa<abstract::Shape>() || b_shp->isa<abstract::NoShape>()) { in GetOutputDetailShape()1185 return b_shp; in GetOutputDetailShape()2096 auto b_shp = (*tuple_shape)[index]; in IsNodeInputDynamicShape() local2097 if (!b_shp->isa<abstract::Shape>()) { in IsNodeInputDynamicShape()2100 if (AnfUtils::IsShapeDynamic(b_shp->cast<abstract::ShapePtr>())) { in IsNodeInputDynamicShape()
136 std::vector<int64_t> b_shp{128}; in TEST_F() local141 auto b_abstract = std::make_shared<abstract::AbstractTensor>(kFloat16, b_shp); in TEST_F()