Searched refs:FORMAT_NCHW_VECT_C (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 50 FORMAT_NCHW_VECT_C = 2, enumerator
118 case FORMAT_NCHW_VECT_C: in GetTensorSpatialDims()
144 case FORMAT_NCHW_VECT_C: in GetTensorDimsFromSpatialDims()
166 case FORMAT_NCHW_VECT_C: in GetTensorBatchDimIndex()
191 case FORMAT_NCHW_VECT_C: in GetTensorFeatureDimIndex()
201 DCHECK_EQ(format, FORMAT_NCHW_VECT_C); in GetTensorInnerFeatureDimIndex()
225 case FORMAT_NCHW_VECT_C: in GetTensorSpatialDimIndex()
317 } else if (format == FORMAT_NCHW || format == FORMAT_NCHW_VECT_C) { in GetTensorDimIndex()
528 if (format == FORMAT_NCHW_VECT_C) { in ShapeFromFormat()
586 (src_format == FORMAT_NCHW_VECT_C ? 4 : 1); in ShapeFromFormat()
|
| D | tensor_format.cc | 46 case FORMAT_NCHW_VECT_C: in ToString()
84 *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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.cc | 62 (data_format == FORMAT_NCHW_VECT_C ? 4 : 1); in PoolParameters()
214 case 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 | spacetodepth_op.cc | 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
|
| D | pooling_ops_common.h | 299 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
|
| D | maxpooling_op.cc | 1161 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 316 if (tensor_format == FORMAT_NCHW_VECT_C) { in CheckFormatConstraintsOnShape()
337 if (format == FORMAT_NCHW_VECT_C) { in MakeShapeFromFormat()
368 if (format == FORMAT_NCHW_VECT_C) { in DimensionsFromShape()
393 if (format == tensorflow::FORMAT_NCHW_VECT_C) { in ShapeFromDimensions()
730 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in AvgPoolShape()
880 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolShape()
950 const int rank = (data_format == FORMAT_NCHW_VECT_C) ? 5 : 4; in MaxPoolV2Shape()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 130 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in FusedConv2DBiasActivationOp()
579 is_int8x4 ? FORMAT_NCHW_VECT_C : FORMAT_NCHW; in launch()
|