Searched refs:kOpFormat_ND (Results 1 – 11 of 11) sorted by relevance
128 const std::set<std::string> kCommonFormatSet = {kOpFormat_DEFAULT, kOpFormat_ND, kOpFormat_NCHW, kO…
64 …put_format = AnfAlgo::GetCNodeName(trans_node) == kTransDataOpName ? kOpFormat_NCHW : kOpFormat_ND; in SetTransNodeAttr()67 …put_format = AnfAlgo::GetCNodeName(trans_node) == kTransDataOpName ? kOpFormat_NCHW : kOpFormat_ND; in SetTransNodeAttr()
573 constexpr auto kOpFormat_ND = "ND"; variable596 kOpFormat_ND,616 const std::set<std::string> kDefaultCompatibleFormat = {kOpFormat_ND, kOpFormat_NCHW, kOpFormat_NHW…
67 {kOpFormat_ND, ge::Format::FORMAT_ND}, in GetGeFormat()
34 auto ge_tensor = TransformUtil::ConvertTensor(me_tensor, kOpFormat_ND); in ConvertAnyUtil()
98 } else if (format == kOpFormat_ND) { in ConvertFormat()
57 if (format == kOpFormat_ND || format == kOpFormat_DEFAULT) { in GetFakeAbstractShape()
274 if (format == kOpFormat_ND || format == kOpFormat_DEFAULT) { in DynamicShapeRangeTrans()
377 return origin_shape_size == kNCHWShapeSize ? kOpFormat_NCHW : kOpFormat_ND; in FormatPass()
1131 format = ori_shape.size() == kNCHWShapeSize ? kOpFormat_NCHW : kOpFormat_ND; in GenDescJson()
989 if (format == kOpFormat_ND || format == kOpFormat_DEFAULT) { in TransShapeToDevice()1036 if (format == kOpFormat_ND || format == kOpFormat_DEFAULT || format == kOpFormat_NCHW) { in TransShapeToDevice()