Home
last modified time | relevance | path

Searched refs:FORMAT_NHWC_VECT_W (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h55 FORMAT_NHWC_VECT_W = 3, enumerator
119 case FORMAT_NHWC_VECT_W: in GetTensorSpatialDims()
145 case FORMAT_NHWC_VECT_W: in GetTensorDimsFromSpatialDims()
167 case FORMAT_NHWC_VECT_W: in GetTensorBatchDimIndex()
187 case FORMAT_NHWC_VECT_W: in GetTensorFeatureDimIndex()
207 DCHECK_EQ(format, FORMAT_NHWC_VECT_W); in GetTensorInnerWidthDimIndex()
222 case FORMAT_NHWC_VECT_W: in GetTensorSpatialDimIndex()
303 if (format == FORMAT_NHWC || format == FORMAT_NHWC_VECT_W) { in GetTensorDimIndex()
516 if (format == FORMAT_NHWC_VECT_W && in ShapeFromFormat()
595 if (src_format == FORMAT_NHWC_VECT_W) { in ShapeFromFormat()
Dtensor_format.cc48 case FORMAT_NHWC_VECT_W: in ToString()
88 *format = FORMAT_NHWC_VECT_W; in FormatFromString()
Dtensor_format_test.cc30 EnumStringPair(FORMAT_NCHW_VECT_C), EnumStringPair(FORMAT_NHWC_VECT_W),
118 format == FORMAT_NHWC_VECT_W) ? DimMaps::kTdmNHWC[num_spatial_dims] : in GetTensorDimMap()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc340 } else if (format == FORMAT_NHWC_VECT_W) { in MakeShapeFromFormat()