Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h42 FORMAT_NCHW = 1, enumerator
119 case FORMAT_NCHW: in GetTensorSpatialDims()
148 case FORMAT_NCHW: in GetTensorDimsFromSpatialDims()
176 case FORMAT_NCHW: in GetTensorBatchDimIndex()
201 case FORMAT_NCHW: in GetTensorFeatureDimIndex()
235 case FORMAT_NCHW: in GetTensorSpatialDimIndex()
328 } else if (format == FORMAT_NCHW || format == FORMAT_NCHW_VECT_C) { in GetTensorDimIndex()
Dtensor_format.cc44 case FORMAT_NCHW: in ToString()
82 *format = FORMAT_NCHW; in FormatFromString()
Dtensor_format_test.cc29 EnumStringPair(FORMAT_NHWC), EnumStringPair(FORMAT_NCHW),
119 (format == FORMAT_NCHW || in GetTensorDimMap()
Dmkl_util.h163 FORMAT_NCHW = 1, enumerator
200 } else if (format == MklTensorFormat::FORMAT_NCHW) {
556 TensorFormat data_format = FORMAT_NCHW; in IsTensorInNCHWFormat()
1011 if (format == MklTensorFormat::FORMAT_NCHW) return memory::format_tag::nchw; in MklTensorFormatToMklDnnDataFormat()
1026 if (format == FORMAT_NCHW) return MklTensorFormat::FORMAT_NCDHW; in TFDataFormatToMklDnn3DDataFormat()
1038 if (format == FORMAT_NCHW) return MklTensorFormat::FORMAT_NCHW; in TFDataFormatToMklDnnDataFormat()
1052 if (format == MklTensorFormat::FORMAT_NCHW || in MklDnnDataFormatToTFDataFormat()
1054 return FORMAT_NCHW; in MklDnnDataFormatToTFDataFormat()
Dmkl_types.h94 #define MKL_TENSOR_FORMAT_NCHW MklTensorFormat::FORMAT_NCHW
/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.cc120 functor::DepthToSpaceOpFunctor<Device, int32, FORMAT_NCHW> functor; in Compute()
124 } else if (data_format_ == FORMAT_NCHW) { in Compute()
125 functor::DepthToSpaceOpFunctor<Device, T, FORMAT_NCHW> functor; in Compute()
178 struct DepthToSpaceOpFunctor<CPUDevice, T, FORMAT_NCHW> {
Dspacetodepth_op.cc135 functor::SpaceToDepthOpFunctor<Device, int32, FORMAT_NCHW> functor; in Compute()
138 } else if (data_format_ == FORMAT_NCHW) { in Compute()
140 functor::SpaceToDepthOpFunctor<Device, RT, FORMAT_NCHW> functor; in Compute()
194 struct SpaceToDepthOpFunctor<CPUDevice, T, FORMAT_NCHW> {
Ddepthtospace_op_gpu.cu.cc177 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> {
242 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>;
247 FORMAT_NCHW>;
252 template struct functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW>;
Dfused_batch_norm_op.cu.cc131 static_assert(tensor_format == FORMAT_NHWC || tensor_format == FORMAT_NCHW,
378 } else if (tensor_format == FORMAT_NCHW) { in operator ()()
383 LAUNCH(FORMAT_NCHW, false, Activation::kIdentity, c, inner); in operator ()()
385 LAUNCH(FORMAT_NCHW, false, Activation::kRelu, c, inner); in operator ()()
387 LAUNCH(FORMAT_NCHW, true, Activation::kIdentity, c, inner); in operator ()()
389 LAUNCH(FORMAT_NCHW, true, Activation::kRelu, c, inner); in operator ()()
Dcudnn_pooling_gpu.cc51 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
65 ShapeFromFormat(FORMAT_NCHW, out_shape, data_format), in Compute()
152 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
163 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
Dbias_op.cc59 } else if (data_format == FORMAT_NCHW) { in GetBiasValueDims()
112 if (data_format_ == FORMAT_NCHW) { in Compute()
133 if (data_format_ == FORMAT_NCHW) { in Compute()
259 if (data_format_ == FORMAT_NCHW) { in Compute()
454 data_format_ = FORMAT_NCHW; in BiasGradOp()
472 if (data_format_ == FORMAT_NCHW) { in ComputeWithReduceSum()
Dpooling_ops_common.cc247 ShapeFromFormat(FORMAT_NCHW, tensor_in.shape(), in Compute()
260 ShapeFromFormat(FORMAT_NCHW, tensor_out_shape, in Compute()
275 case FORMAT_NCHW: in Compute()
475 ShapeFromFormat(FORMAT_NCHW, tensor_in_shape, data_format); in Compute()
486 ShapeFromFormat(FORMAT_NCHW, out_backprop.shape(), data_format); in Compute()
521 transformed_input_data_format = FORMAT_NCHW; in Compute()
560 case FORMAT_NCHW: in Compute()
Dfused_batch_norm_op.cc121 if (tensor_format == FORMAT_NCHW) { in operator ()()
217 if (tensor_format == FORMAT_NCHW) { in operator ()()
259 if (tensor_format == FORMAT_NCHW) { in operator ()()
328 if (tensor_format == FORMAT_NCHW) { in operator ()()
351 if (tensor_format == FORMAT_NCHW) { in operator ()()
504 if (tensor_format == FORMAT_NCHW) { in operator ()()
798 : FORMAT_NCHW; in operator ()()
861 } else if (tensor_format == FORMAT_NHWC && compute_format == FORMAT_NCHW) { in operator ()()
974 if (tensor_format == FORMAT_NHWC && compute_format == FORMAT_NCHW) { in operator ()()
1014 : FORMAT_NCHW; in operator ()()
[all …]
Dconv_ops_3d.cc349 : FORMAT_NCHW; in launch()
355 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
358 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
432 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in launch()
460 ShapeFromFormat(FORMAT_NCHW, in_batch, in launch()
633 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
Dlrn_op.cc243 ShapeFromFormat(FORMAT_NCHW, in.shape(), FORMAT_NHWC), in launch()
253 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
536 ShapeFromFormat(FORMAT_NCHW, in_grads.shape(), in launch()
546 ShapeFromFormat(FORMAT_NCHW, in_image.shape(), in launch()
556 ShapeFromFormat(FORMAT_NCHW, out_image.shape(), in launch()
567 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
Dconv_grad_input_ops.cc230 : FORMAT_NCHW; in operator ()()
303 if (compute_data_format == FORMAT_NCHW) { in operator ()()
314 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
540 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
Dconv_ops_fused_impl.h442 params.data_format == FORMAT_NHWC || params.data_format == FORMAT_NCHW,
543 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
622 ShapeFromFormat(FORMAT_NCHW, out_batch, out_rows,
650 FORMAT_NCHW, // compute_data_format
Dconv_grad_ops_3d.cc1279 : FORMAT_NCHW; in Compute()
1334 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1354 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
1555 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
1773 : FORMAT_NCHW; in Compute()
1830 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in Compute()
1844 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
1863 if (data_format_ == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in Compute()
Dconv_ops.cc752 : FORMAT_NCHW; in operator ()()
837 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
841 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths); in operator ()()
929 if (compute_data_format == FORMAT_NCHW) { in operator ()()
1124 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
Ddepthwise_conv_op.cc312 (data_format_ == FORMAT_NCHW || in DepthwiseConv2dNativeOp()
319 use_cudnn_grouped_conv_ = dtype_ == DT_HALF && data_format_ == FORMAT_NCHW; in DepthwiseConv2dNativeOp()
Dconv_grad_filter_ops.cc835 : FORMAT_NCHW; in operator ()()
899 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
920 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
1111 compute_data_format == FORMAT_NCHW ? FORMAT_OIHW : FORMAT_OHWI; in operator ()()
/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/common_runtime/
Dmkl_tfconversion_pass.cc184 data_format == ToString(FORMAT_NCHW))) { in InsertConversionNodeOnEdge()
270 data_format == ToString(FORMAT_NCHW))) { in InsertInputConversionNode()

12