/external/tensorflow/tensorflow/core/util/ |
D | tensor_format.h | 40 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()
|
D | tensor_format.cc | 44 case FORMAT_NCHW: in ToString() 80 *format = FORMAT_NCHW; in FormatFromString()
|
D | tensor_format_test.cc | 29 EnumStringPair(FORMAT_NHWC), EnumStringPair(FORMAT_NCHW), 119 (format == FORMAT_NCHW || in GetTensorDimMap()
|
D | mkl_util.h | 267 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/ |
D | spacetodepth_op_gpu.cu.cc | 174 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>;
|
D | depthtospace_op_gpu.cu.cc | 178 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>;
|
D | depthtospace_op.cc | 120 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()
|
D | cudnn_pooling_gpu.cc | 51 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()
|
D | spacetodepth_op.cc | 122 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()
|
D | pooling_ops_common.cc | 183 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()
|
D | bias_op.cc | 60 } 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()
|
D | fused_batch_norm_op.cc | 277 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 ()()
|
D | conv_ops_3d.cc | 330 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()
|
D | conv_grad_input_ops.cc | 887 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 ()()
|
D | conv_ops_fused_impl.h | 563 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
|
D | mkl_conv_grad_bias_ops.cc | 82 } else if (data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW) { in Compute()
|
D | conv_ops.cc | 744 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 ()()
|
D | conv_grad_filter_ops.cc | 767 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 ()()
|
D | depthwise_conv_op_gpu.h | 618 case FORMAT_NCHW: in LaunchDepthwiseConv2dGPUSmall() 750 case FORMAT_NCHW: 973 case FORMAT_NCHW: 1612 case FORMAT_NCHW: 1737 case FORMAT_NCHW:
|
D | mkl_pooling_ops_common.h | 421 data_format(TensorFormat::FORMAT_NCHW) {} in MklPoolParameters() 561 if (pool_params->data_format == TensorFormat::FORMAT_NCHW) { in AllocateEmptyOutputTensor()
|
D | cwise_ops_test.cc | 191 if (format == FORMAT_NCHW) { in BiasAddGrad()
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | depthtospace_op.cc | 47 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()
|
D | spacetodepth_op.cc | 47 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/ |
D | mkl_tfconversion_pass.cc | 183 data_format == ToString(FORMAT_NCHW))) { in InsertConversionNodeOnEdge() 268 data_format == ToString(FORMAT_NCHW))) { in InsertInputConversionNode()
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 402 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()
|