/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
bias_ops.cc | 33 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), 36 data_format_ = FORMAT_NHWC; 50 int feature_dim = (data_format_ == FORMAT_NHWC) ? input_shape.dims() - 1 69 TensorFormat data_format_; member in class:tensorflow::__anon38728::BiasOp 80 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), 83 data_format_ = FORMAT_NHWC; 94 int feature_dim = (data_format_ == FORMAT_NHWC) 113 TensorFormat data_format_; member in class:tensorflow::__anon38728::BiasAddGradOp
|
depthtospace_op.cc | 29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), 32 OP_REQUIRES(ctx, data_format_ == FORMAT_NCHW || data_format_ == FORMAT_NHWC, 34 ToString(data_format_), 56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_); 57 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format_); 65 if (data_format_ == FORMAT_NHWC) { 162 TensorFormat data_format_; member in class:tensorflow::__anon38738::DepthToSpaceOp
|
spacetodepth_op.cc | 29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), 32 OP_REQUIRES(ctx, data_format_ == FORMAT_NCHW || data_format_ == FORMAT_NHWC, 34 ToString(data_format_), 56 int feature_dim = GetTensorFeatureDimIndex(input_rank, data_format_); 57 int num_spatial_dims = GetTensorSpatialDims(input_rank, data_format_); 65 if (data_format_ == FORMAT_NHWC) { 166 TensorFormat data_format_; member in class:tensorflow::__anon38780::SpaceToDepthOp
|
batch_norm_op.cc | 34 ctx, FormatFromString(data_format_str, &data_format_), 37 (data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW), 39 "Unsupported data format ", ToString(data_format_), 57 GetTensorFeatureDimIndex(input_shape.dims(), data_format_); 96 TensorFormat data_format_; member in class:tensorflow::__anon38725::FusedBatchNormOp 111 ctx, FormatFromString(data_format_str, &data_format_), 114 (data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW), 116 "Unsupported data format ", ToString(data_format_), 196 TensorFormat data_format_; member in class:tensorflow::__anon38725::FusedBatchNormGradOp [all...] |
pooling_ops.cc | 126 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38761::PoolingOp 157 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), 246 data_format_); 256 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), 359 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38761::MaxPoolGradOp 368 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), 424 int depth_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); 443 int dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i); 456 padding_, data_format_, &dims)); 473 stride_int64s, num_spatial_dims_, data_format_); 504 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38761::AvgPoolGradOp [all...] |
conv_ops.cc | 188 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), 199 int batch_dim = GetTensorBatchDimIndex(num_dims(), data_format_); 200 int feature_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); 215 int input_dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i); 272 const int64 dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i); 299 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38736::ConvOp 339 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), 350 int batch_dim = GetTensorBatchDimIndex(num_dims(), data_format_); 351 int feature_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); 366 int input_dim = GetTensorSpatialDimIndex(num_dims(), data_format_, i) 452 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38736::ConvBackpropInputOp 654 TensorFormat data_format_ = FORMAT_NHWC; member in class:tensorflow::__anon38736::ConvBackpropFilterOp [all...] |
/external/tensorflow/tensorflow/core/kernels/ |
bias_op.cc | 90 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 93 data_format_ = FORMAT_NHWC; 110 if (data_format_ == FORMAT_NCHW) { 133 if (data_format_ == FORMAT_NCHW) { 135 GetBiasValueDims(input, data_format_, &batch, &height, &width, &channel); 175 TensorFormat data_format_; member in class:tensorflow::BiasOp 210 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 213 data_format_ = FORMAT_NHWC; 232 GetBiasValueDims(output_backprop, data_format_, &batch, &height, &width, 245 if (data_format_ == FORMAT_NCHW) 282 TensorFormat data_format_; member in class:tensorflow::BiasGradOp 351 TensorFormat data_format_; member in class:tensorflow::BiasOp 408 TensorFormat data_format_; member in class:tensorflow::BiasGradOp [all...] |
depthtospace_op.cc | 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), 59 context, data_format_ == FORMAT_NHWC, 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), 83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); 85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); 87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); 89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * 109 ShapeFromFormat(data_format_, batch_size, output_height, 124 } else if (data_format_ == FORMAT_NCHW) { 132 // NOTE: Assumes data_format_ == FORMAT_NHWC here, since we have rejecte 143 TensorFormat data_format_; member in class:tensorflow::DepthToSpaceOp [all...] |
spacetodepth_op.cc | 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), 59 context, data_format_ == FORMAT_NHWC, 71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), 83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); 85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); 87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); 89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * 110 ShapeFromFormat(data_format_, batch_size, output_height, 126 } else if (data_format_ == FORMAT_NCHW) { 134 // NOTE: Assumes data_format_ == FORMAT_NHWC here, since we have rejecte 145 TensorFormat data_format_; member in class:tensorflow::SpaceToDepthOp [all...] |
avgpooling_op.cc | 55 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 58 context, data_format_ == FORMAT_NHWC, 79 padding_, data_format_, tensor_in.shape()}; 102 TensorFormat data_format_; member in class:tensorflow::AvgPoolingOp 120 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 131 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); 132 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); 141 padding_, data_format_, tensor_in.shape()}; 155 if (data_format_ == FORMAT_NCHW) { 158 stride_, padding_, data_format_, tensor_in, output_shape 176 TensorFormat data_format_; member in class:tensorflow::AvgPoolingOp 350 TensorFormat data_format_; member in class:tensorflow::AvgPoolingGradOp 423 TensorFormat data_format_; member in class:tensorflow::AvgPoolingGradOp 548 TensorFormat data_format_; member in class:tensorflow::AvgPoolingGradOpCustomGPUKernel [all...] |
depthwise_conv_op.cc | 266 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 272 stride_ = GetTensorDim(strides_, data_format_, 'H'); 273 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 274 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 275 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 310 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); 322 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); 330 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); 349 ShapeFromFormat(data_format_, batch, out_rows, out_cols, out_depth); 380 padding_, output, data_format_); 409 TensorFormat data_format_; member in class:tensorflow::DepthwiseConv2dNativeOp [all...] |
mkl_conv_grad_bias_ops.cc | 56 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 79 } else if (data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW) { 80 mkl_context.c_size = GetTensorDim(input, data_format_, 'C'); 105 mkl_context.in_sizes[MklDims::W] = GetTensorDim(input, data_format_, 'W'); 106 mkl_context.in_sizes[MklDims::H] = GetTensorDim(input, data_format_, 'H'); 107 mkl_context.in_sizes[MklDims::C] = GetTensorDim(input, data_format_, 'C'); 108 mkl_context.in_sizes[MklDims::N] = GetTensorDim(input, data_format_, 'N'); 109 GetStridesFromSizes(data_format_, mkl_context.in_strides, 250 TensorFormat data_format_; member in class:tensorflow::MklConv2DCustomBackpropBiasOp [all...] |
pooling_ops_3d.cc | 136 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 140 context, data_format_ == FORMAT_NHWC, 155 (GetTensorDim(ksize_, data_format_, 'N') == 1 && 156 GetTensorDim(stride_, data_format_, 'N') == 1), 160 (GetTensorDim(ksize_, data_format_, 'C') == 1 && 161 GetTensorDim(stride_, data_format_, 'C') == 1), 171 const int64 depth = GetTensorDim(tensor_in, data_format_, 'C'); 172 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); 176 {GetTensorDim(tensor_in, data_format_, '2'), 177 GetTensorDim(tensor_in, data_format_, '1') 203 TensorFormat data_format_; member in class:tensorflow::Pooling3DOp 377 TensorFormat data_format_; member in class:tensorflow::MaxPooling3dGradOp 538 TensorFormat data_format_; member in class:tensorflow::AvgPooling3dGradOp 713 TensorFormat data_format_; member in class:tensorflow::MaxPooling3dGradGradOp [all...] |
pooling_ops_common.h | 85 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 88 context, data_format_ == FORMAT_NHWC, 93 data_format_ = FORMAT_NHWC; 261 TensorFormat data_format_; member in class:tensorflow::MaxPoolingOp 295 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 299 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C, 304 data_format_ = FORMAT_NHWC; 351 padding_, data_format_, tensor_in.shape()}; 508 TensorFormat data_format_; member in class:tensorflow::MaxPoolingV2Op [all...] |
conv_grad_ops_3d.cc | 64 {GetTensorDim(input_shape, data_format_, '0'), \ 65 GetTensorDim(input_shape, data_format_, '1'), \ 66 GetTensorDim(input_shape, data_format_, '2')}}; \ 67 const int64 in_depth = GetTensorDim(input_shape, data_format_, 'C'); \ 71 const int64 output_cols = GetTensorDim(out_backprop, data_format_, '2'); \ 72 const int64 output_rows = GetTensorDim(out_backprop, data_format_, '1'); \ 73 const int64 output_planes = GetTensorDim(out_backprop, data_format_, '0'); \ 79 context, out_depth == GetTensorDim(out_backprop, data_format_, 'C'), \ 83 {GetTensorDim(stride_, data_format_, '0'), \ 84 GetTensorDim(stride_, data_format_, '1'), 228 TensorFormat data_format_; member in class:tensorflow::Conv3DBackpropInputOp 375 TensorFormat data_format_; member in class:tensorflow::Conv3DBackpropFilterOp 758 TensorFormat data_format_; member in class:tensorflow::Conv3DBackpropInputOp 1103 TensorFormat data_format_; member in class:tensorflow::Conv3DBackpropFilterOp [all...] |
conv_ops_3d.cc | 71 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 79 (GetTensorDim(stride_, data_format_, 'N') == 1 && 80 GetTensorDim(stride_, data_format_, 'C') == 1), 103 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); 104 const int64 in_batch = GetTensorDim(input, data_format_, 'N'); 113 {GetTensorDim(input, data_format_, '0'), 114 GetTensorDim(input, data_format_, '1'), 115 GetTensorDim(input, data_format_, '2')}}; 118 std::array<int64, 3> strides = {{GetTensorDim(stride_, data_format_, '0'), 119 GetTensorDim(stride_, data_format_, '1') 140 TensorFormat data_format_; member in class:tensorflow::Conv3DOp [all...] |
conv_ops_using_gemm.cc | 436 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 438 OP_REQUIRES(context, data_format_ == FORMAT_NHWC, 444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 479 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); 490 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); 500 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); 509 const int64 batch_raw = GetTensorDim(input, data_format_, 'N'); 517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W') 558 TensorFormat data_format_; member in class:tensorflow::Conv2DUsingGemmOp [all...] |
maxpooling_op.cc | 203 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 206 context, data_format_ == FORMAT_NHWC, 303 TensorFormat data_format_; member in class:tensorflow::MaxPoolingGradOp 340 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 351 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); 352 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); 399 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N'); 400 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); 408 stride, padding_, data_format_, &tensor_in, &tensor_out, out_backprop, 411 CHECK(data_format_ == FORMAT_NHWC 422 TensorFormat data_format_; member in class:tensorflow::MaxPoolingGradOp 633 TensorFormat data_format_; member in class:tensorflow::MaxPoolingGradGradOp 729 TensorFormat data_format_; member in class:tensorflow::MaxPoolingGradGradOp 788 TensorFormat data_format_; member in class:tensorflow::MaxPoolingNoMaskOp 866 TensorFormat data_format_; member in class:tensorflow::MaxPoolingNoMaskV2Op 975 TensorFormat data_format_; member in class:tensorflow::MaxPoolingGradWithArgmaxOp 1109 TensorFormat data_format_; member in class:tensorflow::MaxPoolingNoMaskOp 1202 TensorFormat data_format_; member in class:tensorflow::MaxPoolingNoMaskV2Op [all...] |
mkl_avgpooling_op.cc | 51 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 81 pool_params.Init(context, ksize_, stride_, padding_, data_format_, 84 pool_params.Init(context, ksize_, stride_, padding_, data_format_, 89 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); 131 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); 214 TensorFormat data_format_; member in class:tensorflow::MklAvgPoolingOp 227 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 258 pool_params.Init(context, ksize_, stride_, padding_, data_format_, 267 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); 322 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); 428 TensorFormat data_format_; member in class:tensorflow::MklAvgPoolingGradOp [all...] |
conv_ops.cc | 264 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 275 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); 276 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); 277 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H'); 278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 287 const int64 dilation_n = GetTensorDim(dilations_, data_format_, 'N'); 288 const int64 dilation_c = GetTensorDim(dilations_, data_format_, 'C'); 289 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); 290 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); 328 const int64 in_depth = GetTensorDim(input, data_format_, 'C') 432 TensorFormat data_format_; member in class:tensorflow::Conv2DOp [all...] |
depthwise_conv_grad_op.cc | 67 const int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); \ 73 const int64 input_cols_raw = GetTensorDim(input_shape, data_format_, 'W'); \ 82 GetTensorDim(out_backprop.shape(), data_format_, 'H'); \ 89 GetTensorDim(out_backprop.shape(), data_format_, 'W'); \ 95 const int64 in_depth = GetTensorDim(input_shape, data_format_, 'C'); \ 101 GetTensorDim(out_backprop.shape(), data_format_, 'C'); \ 534 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 537 stride_ = GetTensorDim(strides_, data_format_, 'H'); 538 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); 539 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N') 589 TensorFormat data_format_; member in class:tensorflow::DepthwiseConv2dNativeBackpropInputOp 967 TensorFormat data_format_; member in class:tensorflow::DepthwiseConv2dNativeBackpropFilterOp [all...] |
mkl_concat_op.cc | 242 TensorFormat data_format_; member in class:tensorflow::MklConcatOp 570 TensorFormat data_format_; member in class:tensorflow::MklConcatOp [all...] |
mkl_conv_grad_filter_ops.cc | 67 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), 71 int stride_n = GetTensorDim(strides_, data_format_, 'N'); 72 int stride_c = GetTensorDim(strides_, data_format_, 'C'); 112 MklSizesToTFSizes(context, data_format_, mkl_context.input_shape, 125 MklSizesToTFSizes(context, data_format_, mkl_context.out_backprop_shape, 135 padding_, data_format_, &backprop_dims)); 178 GetStridesFromSizes(data_format_, mkl_context.in_strides, 180 GetStridesFromSizes(data_format_, mkl_context.out_strides, 424 TensorFormat data_format_; member in class:tensorflow::MklConv2DCustomBackpropFilterOp
|
mkl_conv_ops.h | 62 TensorFormat data_format_; member in class:tensorflow::MklDnnConvUtil 67 : context_(context), strides_(strides), padding_(pad), data_format_(fm) {} 76 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 77 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); 97 int64 input_depth_raw = GetTensorDim(input_shape, data_format_, 'C'); 101 int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); 106 int64 input_cols_raw = GetTensorDim(input_shape, data_format_, 'W'); 111 int64 input_batch_raw = GetTensorDim(input_shape, data_format_, 'N'); 157 int input_depth = GetTensorDim(input_shape, data_format_, 'C'); 224 int input_rows = GetTensorDim(input_shape, data_format_, 'H') 554 TensorFormat data_format_; member in class:tensorflow::MklConv2DBackpropCommonOp [all...] |
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
fused_conv2d_bias_activation_op.cc | 94 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), 108 stride_rows_ = GetTensorDim(strides, data_format_, 'H'); 109 stride_cols_ = GetTensorDim(strides, data_format_, 'W'); 112 (GetTensorDim(strides, data_format_, 'N') == 1 && 113 GetTensorDim(strides, data_format_, 'C') == 1), 123 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), 197 const int32 batch_size = GetTensorDim(conv_input, data_format_, 'N'); 198 const int32 conv_input_rows = GetTensorDim(conv_input, data_format_, 'H'); 199 const int32 conv_input_cols = GetTensorDim(conv_input, data_format_, 'W'); 208 // Initialize the output tensor shape according to data_format_ 241 TensorFormat data_format_; member in class:tensorflow::FusedConv2DBiasActivationOp [all...] |