Home
last modified time | relevance | path

Searched refs:FORMAT_NCHW_VECT_C (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h52 FORMAT_NCHW_VECT_C = 2, enumerator
123 case FORMAT_NCHW_VECT_C: in GetTensorSpatialDims()
152 case FORMAT_NCHW_VECT_C: in GetTensorDimsFromSpatialDims()
177 case FORMAT_NCHW_VECT_C: in GetTensorBatchDimIndex()
202 case FORMAT_NCHW_VECT_C: in GetTensorFeatureDimIndex()
212 DCHECK_EQ(format, FORMAT_NCHW_VECT_C); in GetTensorInnerFeatureDimIndex()
236 case FORMAT_NCHW_VECT_C: in GetTensorSpatialDimIndex()
328 } else if (format == FORMAT_NCHW || format == FORMAT_NCHW_VECT_C) { in GetTensorDimIndex()
541 if (format == FORMAT_NCHW_VECT_C) { in ShapeFromFormat()
599 (src_format == FORMAT_NCHW_VECT_C ? 4 : 1); in ShapeFromFormat()
Dtensor_format.cc46 case FORMAT_NCHW_VECT_C: in ToString()
86 *format = FORMAT_NCHW_VECT_C; in FormatFromString()
Dtensor_format_test.cc30 EnumStringPair(FORMAT_NCHW_VECT_C), EnumStringPair(FORMAT_NHWC_VECT_W),
120 format == FORMAT_NCHW_VECT_C) ? DimMaps::kTdmNCHW[num_spatial_dims] : in GetTensorDimMap()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc46 if (data_format == FORMAT_NCHW_VECT_C) { in Compile()
171 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
Dspacetodepth_op.cc46 if (data_format == FORMAT_NCHW_VECT_C) { in Compile()
175 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
Dpooling_ops.cc163 data_format_ != FORMAT_NCHW_VECT_C && in MaxPoolOp()
/external/tensorflow/tensorflow/core/kernels/
Dspacetodepth_op.cc84 const bool is_int8x4 = (data_format_ == FORMAT_NCHW_VECT_C); in Compute()
131 if (data_format_ == FORMAT_NCHW_VECT_C) { in Compute()
Ddepthtospace_op.cc71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
Dpooling_ops_common.cc134 (data_format == FORMAT_NCHW_VECT_C ? 4 : 1); in PoolParameters()
278 case FORMAT_NCHW_VECT_C: in Compute()
Dpooling_ops_common.h344 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
Dmaxpooling_op.cc1206 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc462 if (tensor_format == FORMAT_NCHW_VECT_C) { in CheckFormatConstraintsOnShape()
503 if (format == FORMAT_NCHW_VECT_C) { in MakeShapeFromFormat()
535 if (format == FORMAT_NCHW_VECT_C) { in DimensionsFromShape()
560 if (format == tensorflow::FORMAT_NCHW_VECT_C) { in ShapeFromDimensions()
1064 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in AvgPoolShape()
1505 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolShapeImpl()
1612 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolV2Shape()