HomeSort by relevance Sort by last modified time
    Searched defs:data_format_ (Results 1 - 25 of 29) sorted by null

1 2

  /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...]

Completed in 707 milliseconds

1 2