Searched refs:FORMAT_OIHW_VECT_I (Results 1 – 5 of 5) sorted by relevance
91 FORMAT_OIHW_VECT_I = 2, enumerator127 if (format == FORMAT_OIHW_VECT_I) { in GetFilterTensorSpatialDims()154 if (format == FORMAT_OIHW_VECT_I) { in GetFilterTensorDimsFromSpatialDims()244 case FORMAT_OIHW_VECT_I: in GetFilterTensorSpatialDimIndex()255 DCHECK_EQ(format, FORMAT_OIHW_VECT_I); in GetFilterTensorInnerInputChannelsDimIndex()268 case FORMAT_OIHW_VECT_I: in GetFilterTensorInputChannelsDimIndex()283 case FORMAT_OIHW_VECT_I: in GetFilterTensorOutputChannelsDimIndex()387 filter_tensor_format == FORMAT_OIHW_VECT_I) { in GetFilterDimIndex()552 if (format == FORMAT_OIHW_VECT_I) { in ShapeFromFilterTensorFormat()613 (src_filter_format == FORMAT_OIHW_VECT_I ? 4 : 1); in ShapeFromFilterFormat()
66 case FORMAT_OIHW_VECT_I: in ToString()113 *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()
134 OP_REQUIRES(context, (is_int8x4 == (filter_format_ == FORMAT_OIHW_VECT_I)), in FusedConv2DBiasActivationOp()
483 if (filter_format == FORMAT_OIHW_VECT_I) { in Conv2DShapeImpl()