Searched refs:ShapeSpecificationType (Results 1 – 3 of 3) sorted by relevance
33 static std::vector<ShapeSpecificationType> _range_;37 std::vector<ShapeSpecificationType>38 ReshapeOpTest<ShapeSpecificationType>::_range_{39 ShapeSpecificationType::kAsReshapeOption,40 ShapeSpecificationType::kAsConstantTensor,41 ShapeSpecificationType::kAsTensor};47 for (ShapeSpecificationType shape_type : in TYPED_TEST()48 ReshapeOpTest<ShapeSpecificationType>::_range_) { in TYPED_TEST()49 if (shape_type == ShapeSpecificationType::kAsTensor) { in TYPED_TEST()65 for (ShapeSpecificationType shape_type : in TYPED_TEST()[all …]
31 enum class ShapeSpecificationType { enum50 ShapeSpecificationType shape_type) { in ReshapeOpModel()52 case ShapeSpecificationType::kAsTensor: in ReshapeOpModel()55 case ShapeSpecificationType::kAsConstantTensor: in ReshapeOpModel()59 case ShapeSpecificationType::kAsReshapeOption: in ReshapeOpModel()
30 std::vector<ShapeSpecificationType> shape_types = { in TYPED_TEST()31 ShapeSpecificationType::kAsReshapeOption, in TYPED_TEST()32 ShapeSpecificationType::kAsConstantTensor}; in TYPED_TEST()34 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST()45 std::vector<ShapeSpecificationType> shape_types = { in TYPED_TEST()46 ShapeSpecificationType::kAsReshapeOption, in TYPED_TEST()47 ShapeSpecificationType::kAsConstantTensor}; in TYPED_TEST()49 for (ShapeSpecificationType shape_type : shape_types) { in TYPED_TEST()