Searched refs:FORMAT_NCHW_VECT_C (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 52 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()
542 if (format == FORMAT_NCHW_VECT_C) { in ShapeFromFormat()
600 (src_format == FORMAT_NCHW_VECT_C ? 4 : 1); in ShapeFromFormat()
|
| D | tensor_format.cc | 46 case FORMAT_NCHW_VECT_C: in ToString()
86 *format = FORMAT_NCHW_VECT_C; in FormatFromString()
|
| D | tensor_format_test.cc | 30 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/ |
| D | depthtospace_op.cc | 46 if (data_format == FORMAT_NCHW_VECT_C) { in Compile()
171 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
|
| D | spacetodepth_op.cc | 46 if (data_format == FORMAT_NCHW_VECT_C) { in Compile()
175 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
|
| D | pooling_ops.cc | 194 if (data_format_ == FORMAT_NCHW_VECT_C) { in Compile()
209 data_format_ == FORMAT_NCHW_VECT_C ? FORMAT_NCHW : data_format_, in Compile()
212 if (data_format_ == FORMAT_NCHW_VECT_C) { in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetodepth_op.cc | 84 const bool is_int8x4 = (data_format_ == FORMAT_NCHW_VECT_C); in Compute()
131 if (data_format_ == FORMAT_NCHW_VECT_C) { in Compute()
|
| D | depthtospace_op.cc | 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
|
| D | pooling_ops_common.cc | 135 (data_format == FORMAT_NCHW_VECT_C ? 4 : 1); in PoolParameters()
281 case FORMAT_NCHW_VECT_C: in Compute()
|
| D | pooling_ops_common.h | 343 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
|
| D | maxpooling_op.cc | 1280 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 462 if (tensor_format == FORMAT_NCHW_VECT_C) { in CheckFormatConstraintsOnShape()
506 if (format == FORMAT_NCHW_VECT_C) { in MakeShapeFromFormat()
539 if (format == FORMAT_NCHW_VECT_C) { in DimensionsFromShape()
567 if (format == tensorflow::FORMAT_NCHW_VECT_C) { in ShapeFromDimensions()
745 if (data_format == FORMAT_NCHW_VECT_C) { in Conv2DShapeImpl()
1119 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in AvgPoolShape()
1596 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolShapeImpl()
1703 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolV2Shape()
|