Searched refs:FORMAT_OIHW_VECT_I (Results 1 – 4 of 4) sorted by relevance
96 FORMAT_OIHW_VECT_I = 3, enumerator135 if (format == FORMAT_OIHW_VECT_I) { in GetFilterTensorSpatialDims()165 if (format == FORMAT_OIHW_VECT_I) { in GetFilterTensorDimsFromSpatialDims()255 case FORMAT_OIHW_VECT_I: in GetFilterTensorSpatialDimIndex()266 DCHECK_EQ(format, FORMAT_OIHW_VECT_I); in GetFilterTensorInnerInputChannelsDimIndex()279 case FORMAT_OIHW_VECT_I: in GetFilterTensorInputChannelsDimIndex()294 case FORMAT_OIHW_VECT_I: in GetFilterTensorOutputChannelsDimIndex()398 filter_tensor_format == FORMAT_OIHW_VECT_I) { in GetFilterDimIndex()565 if (format == FORMAT_OIHW_VECT_I) { in ShapeFromFilterTensorFormat()626 (src_filter_format == FORMAT_OIHW_VECT_I ? 4 : 1); in ShapeFromFilterFormat()
68 case FORMAT_OIHW_VECT_I: in ToString()115 *format = FORMAT_OIHW_VECT_I; in FilterFormatFromString()
37 EnumStringPair(FORMAT_OIHW_VECT_I),132 format == FORMAT_OIHW_VECT_I) ? DimMaps::kFdmOIHW[num_spatial_dims] in GetFilterDimMap()
650 if (filter_format == FORMAT_OIHW_VECT_I) { in Conv2DShapeImpl()