Searched refs:GeFormat (Results 1 – 5 of 5) sorted by relevance
81 GeFormat TransformUtil::ConvertFormat(const string &format) { in ConvertFormat()83 return GeFormat::FORMAT_NCHW; in ConvertFormat()85 return GeFormat::FORMAT_NDHWC; in ConvertFormat()87 return GeFormat::FORMAT_NCDHW; in ConvertFormat()89 return GeFormat::FORMAT_DHWNC; in ConvertFormat()91 return GeFormat::FORMAT_DHWCN; in ConvertFormat()93 return GeFormat::FORMAT_NC1HWC0; in ConvertFormat()95 return GeFormat::FORMAT_NHWC; in ConvertFormat()97 return GeFormat::FORMAT_HWCN; in ConvertFormat()99 return GeFormat::FORMAT_ND; in ConvertFormat()[all …]
45 using GeFormat = ge::Format; variable
58 static GeFormat ConvertFormat(const std::string &format);
636 ASSERT_EQ(ge_tensor_ptr->GetTensorDesc().GetFormat(), GeFormat::FORMAT_NCHW); in TEST_F()676 ASSERT_EQ(TransformUtil::ConvertFormat(kOpFormat_NCHW), GeFormat::FORMAT_NCHW); in TEST_F()677 ASSERT_EQ(TransformUtil::ConvertFormat(kOpFormat_NC1HWC0), GeFormat::FORMAT_NC1HWC0); in TEST_F()678 ASSERT_EQ(TransformUtil::ConvertFormat(kOpFormat_NHWC), GeFormat::FORMAT_NHWC); in TEST_F()679 ASSERT_EQ(TransformUtil::ConvertFormat("xyz"), GeFormat::FORMAT_ND); in TEST_F()
111 class GeFormat(enum.IntEnum): class570 attr_value = GeFormat.get_format_name(int(attr_value))