Searched refs:FORMAT_NHWC_VECT_W (Results 1 – 5 of 5) sorted by relevance
57 FORMAT_NHWC_VECT_W = 3, enumerator124 case FORMAT_NHWC_VECT_W: in GetTensorSpatialDims()153 case FORMAT_NHWC_VECT_W: in GetTensorDimsFromSpatialDims()178 case FORMAT_NHWC_VECT_W: in GetTensorBatchDimIndex()198 case FORMAT_NHWC_VECT_W: in GetTensorFeatureDimIndex()218 DCHECK_EQ(format, FORMAT_NHWC_VECT_W); in GetTensorInnerWidthDimIndex()233 case FORMAT_NHWC_VECT_W: in GetTensorSpatialDimIndex()314 if (format == FORMAT_NHWC || format == FORMAT_NHWC_VECT_W) { in GetTensorDimIndex()529 if (format == FORMAT_NHWC_VECT_W && in ShapeFromFormat()608 if (src_format == FORMAT_NHWC_VECT_W) { in ShapeFromFormat()
48 case FORMAT_NHWC_VECT_W: in ToString()90 *format = FORMAT_NHWC_VECT_W; in FormatFromString()
30 EnumStringPair(FORMAT_NCHW_VECT_C), EnumStringPair(FORMAT_NHWC_VECT_W),118 format == FORMAT_NHWC_VECT_W) ? DimMaps::kTdmNHWC[num_spatial_dims] : in GetTensorDimMap()
164 data_format_ != FORMAT_NHWC_VECT_W, in MaxPoolOp()
506 } else if (format == FORMAT_NHWC_VECT_W) { in MakeShapeFromFormat()