Home
last modified time | relevance | path

Searched refs:FORMAT_NCHW (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h40 FORMAT_NCHW = 1, enumerator
114 case FORMAT_NCHW: in GetTensorSpatialDims()
140 case FORMAT_NCHW: in GetTensorDimsFromSpatialDims()
165 case FORMAT_NCHW: in GetTensorBatchDimIndex()
190 case FORMAT_NCHW: in GetTensorFeatureDimIndex()
224 case FORMAT_NCHW: in GetTensorSpatialDimIndex()
317 } else if (format == FORMAT_NCHW || format == FORMAT_NCHW_VECT_C) { in GetTensorDimIndex()
Dtensor_format.cc44 case FORMAT_NCHW: in ToString()
80 *format = FORMAT_NCHW; in FormatFromString()
Dtensor_format_test.cc29 EnumStringPair(FORMAT_NHWC), EnumStringPair(FORMAT_NCHW),
119 (format == FORMAT_NCHW || in GetTensorDimMap()
Dmkl_util.h267 TensorFormat data_format = FORMAT_NCHW; in IsTensorInNCHWFormat()
735 TensorFormat data_format = FORMAT_NCHW; in IsTensorInNCHWFormat()
1453 else if (format == FORMAT_NCHW) in TFDataFormatToMklDnn3DDataFormat()
1467 else if (format == FORMAT_NCHW) in TFDataFormatToMklDnnDataFormat()
1482 return FORMAT_NCHW; in MklDnnDataFormatToTFDataFormat()
/external/tensorflow/tensorflow/core/kernels/
Dspacetodepth_op_gpu.cu.cc174 struct SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NCHW> {
239 template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NCHW>;
244 FORMAT_NCHW>;
249 template struct functor::SpaceToDepthOpFunctor<GPUDevice, uint8, FORMAT_NCHW>;
253 template struct functor::SpaceToDepthOpFunctor<GPUDevice, int32, FORMAT_NCHW>;
Ddepthtospace_op_gpu.cu.cc178 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> {
243 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>;
248 FORMAT_NCHW>;
253 template struct functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW>;
Ddepthtospace_op.cc120 functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW> functor; in Compute()
124 } else if (data_format_ == FORMAT_NCHW) { in Compute()
125 functor::DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> functor; in Compute()
Dcudnn_pooling_gpu.cc51 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
65 ShapeFromFormat(FORMAT_NCHW, out_shape, data_format), in Compute()
137 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
148 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
Dspacetodepth_op.cc122 functor::SpaceToDepthOpFunctor<GPUDevice, int32, FORMAT_NCHW> functor; in Compute()
126 } else if (data_format_ == FORMAT_NCHW) { in Compute()
127 functor::SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NCHW> functor; in Compute()
Dpooling_ops_common.cc183 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
196 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
211 case FORMAT_NCHW: in Compute()
315 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
326 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
Dbias_op.cc60 } else if (data_format == FORMAT_NCHW) { in GetBiasValueDims()
113 if (data_format_ == FORMAT_NCHW) { in Compute()
134 if (data_format_ == FORMAT_NCHW) { in Compute()
273 if (data_format_ == FORMAT_NCHW) { in Compute()
486 data_format_ = FORMAT_NCHW; in BiasGradOp()
504 if (data_format_ == FORMAT_NCHW) { in ComputeWithReduceSum()
Dfused_batch_norm_op.cc277 if (tensor_format == FORMAT_NCHW) { in operator ()()
282 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
293 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
416 if (tensor_format == FORMAT_NCHW) { in operator ()()
422 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
434 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
446 ShapeFromFormat(FORMAT_NCHW, batch_size, in operator ()()
Dconv_ops_3d.cc330 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
399 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
421 FORMAT_NCHW, in launch()
Dconv_grad_input_ops.cc887 FORMAT_NCHW, dims.batch_size, dims.spatial_dims[0].output_size, in operator ()()
909 FORMAT_NCHW, in operator ()()
937 FORMAT_NCHW, // compute_data_format in operator ()()
1015 ShapeFromFormat(FORMAT_NCHW, in operator ()()
1034 To32Bit(in_backprop_remove_padding.tensor<T, 4>()), FORMAT_NCHW); in operator ()()
Dconv_ops_fused_impl.h563 params.data_format == FORMAT_NHWC || params.data_format == FORMAT_NCHW,
649 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
728 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
757 FORMAT_NCHW, // compute_data_format
Dmkl_conv_grad_bias_ops.cc82 } else if (data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW) { in Compute()
Dconv_ops.cc744 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
805 ShapeFromFormat(FORMAT_NCHW, out_batch, in operator ()()
833 FORMAT_NCHW, // compute_data_format in operator ()()
Dconv_grad_filter_ops.cc767 FORMAT_NCHW, dims.batch_size, dims.spatial_dims[0].output_size, in operator ()()
787 FORMAT_NCHW, GetTensorDim(compatible_input, data_format, 'N'), in operator ()()
825 FORMAT_NCHW, // compute_data_format in operator ()()
Ddepthwise_conv_op_gpu.h618 case FORMAT_NCHW: in LaunchDepthwiseConv2dGPUSmall()
750 case FORMAT_NCHW:
973 case FORMAT_NCHW:
1612 case FORMAT_NCHW:
1737 case FORMAT_NCHW:
Dmkl_pooling_ops_common.h421 data_format(TensorFormat::FORMAT_NCHW) {} in MklPoolParameters()
561 if (pool_params->data_format == TensorFormat::FORMAT_NCHW) { in AllocateEmptyOutputTensor()
Dcwise_ops_test.cc191 if (format == FORMAT_NCHW) { in BiasAddGrad()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc47 data_format = FORMAT_NCHW; in Compile()
53 OP_REQUIRES(ctx, data_format == FORMAT_NCHW || data_format == FORMAT_NHWC, in Compile()
171 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
Dspacetodepth_op.cc47 data_format = FORMAT_NCHW; in Compile()
53 OP_REQUIRES(ctx, data_format == FORMAT_NCHW || data_format == FORMAT_NHWC, in Compile()
175 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
/external/tensorflow/tensorflow/core/graph/
Dmkl_tfconversion_pass.cc183 data_format == ToString(FORMAT_NCHW))) { in InsertConversionNodeOnEdge()
268 data_format == ToString(FORMAT_NCHW))) { in InsertInputConversionNode()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc402 ShapeFromFormat(FORMAT_NCHW, batch_size, rows, cols, depth); in TransformNHWCToNCHW()
516 auto pad_data_format = is_int8x4 ? FORMAT_NCHW : data_format; in launch()
566 FORMAT_NCHW, batch_size, output_rows, output_cols, output_depth); in launch()
579 is_int8x4 ? FORMAT_NCHW_VECT_C : FORMAT_NCHW; in launch()

12