Home
last modified time | relevance | path

Searched refs:data_format_ (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h62 TensorFormat data_format_; variable
72 data_format_(fm) {} in context_()
82 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder()
83 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder()
86 int stride_planes = GetTensorDim(strides_, data_format_, '0'); in GetStridesInMklOrder()
87 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder()
88 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder()
99 int dilations_rows = GetTensorDim(dilations_, data_format_, 'H'); in GetDilationsInMklOrder()
100 int dilations_cols = GetTensorDim(dilations_, data_format_, 'W'); in GetDilationsInMklOrder()
103 int dilations_planes = GetTensorDim(dilations_, data_format_, '0'); in GetDilationsInMklOrder()
[all …]
Dmkl_conv_ops.cc443 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklConvOp()
449 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp()
450 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp()
467 const int64 dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in MklConvOp()
468 const int64 dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in MklConvOp()
469 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp()
470 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp()
483 (GetTensorDim(dilations_, data_format_, 'N') == 1 && in MklConvOp()
484 GetTensorDim(dilations_, data_format_, 'C') == 1), in MklConvOp()
490 (GetTensorDim(dilations_, data_format_, '0') > 0 && in MklConvOp()
[all …]
Dmkl_conv_grad_input_ops.cc335 diff_dst_tensor.shape(), this->data_format_, in Compute()
376 this->data_format_, this->dilations_); in Compute()
389 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
390 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
Dmkl_conv_grad_filter_ops.cc424 this->data_format_, this->dilations_); in Compute()
434 ? TFDataFormatToMklDnnDataFormat(this->data_format_) in Compute()
435 : TFDataFormatToMklDnn3DDataFormat(this->data_format_); in Compute()
457 depth = (this->data_format_ == FORMAT_NCHW) in Compute()
Dmkl_concat_op.cc442 TensorFormat data_format_; member in tensorflow::MklConcatOp
451 data_format_(TensorFormat::FORMAT_NCHW), in MklConcatOp()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d.cc131 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Pooling3DOp()
135 context, data_format_ == FORMAT_NHWC, in Pooling3DOp()
150 (GetTensorDim(ksize_, data_format_, 'N') == 1 && in Pooling3DOp()
151 GetTensorDim(stride_, data_format_, 'N') == 1), in Pooling3DOp()
155 (GetTensorDim(ksize_, data_format_, 'C') == 1 && in Pooling3DOp()
156 GetTensorDim(stride_, data_format_, 'C') == 1), in Pooling3DOp()
166 const int64 depth = GetTensorDim(tensor_in, data_format_, 'C'); in Compute()
167 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); in Compute()
171 {GetTensorDim(tensor_in, data_format_, '2'), in Compute()
172 GetTensorDim(tensor_in, data_format_, '1'), in Compute()
[all …]
Dconv_grad_ops_3d.cc178 data_format_(FORMAT_NHWC), in Conv3DBackpropInputOp()
184 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
187 context, data_format_ == FORMAT_NHWC, in Conv3DBackpropInputOp()
197 (GetTensorDim(dilation_, data_format_, 'C') == 1 && in Conv3DBackpropInputOp()
198 GetTensorDim(dilation_, data_format_, 'N') == 1), in Conv3DBackpropInputOp()
205 (GetTensorDim(dilation_, data_format_, '0') == 1 && in Conv3DBackpropInputOp()
206 GetTensorDim(dilation_, data_format_, '1') == 1 && in Conv3DBackpropInputOp()
207 GetTensorDim(dilation_, data_format_, '2') == 1), in Conv3DBackpropInputOp()
218 (GetTensorDim(stride_, data_format_, 'C') == 1 && in Conv3DBackpropInputOp()
219 GetTensorDim(stride_, data_format_, 'N') == 1), in Conv3DBackpropInputOp()
[all …]
Dbias_op.cc92 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
95 data_format_ = FORMAT_NHWC; in BiasOp()
112 if (data_format_ == FORMAT_NCHW) { in Compute()
133 if (data_format_ == FORMAT_NCHW) { in Compute()
135 GetBiasValueDims(input, data_format_, &batch, &height, &width, &depth, in Compute()
202 TensorFormat data_format_; member in tensorflow::BiasOp
223 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
226 data_format_ = FORMAT_NHWC; in BiasGradOp()
245 GetBiasValueDims(output_backprop, data_format_, &batch, &height, &width, in Compute()
259 if (data_format_ == FORMAT_NCHW) { in Compute()
[all …]
Davgpooling_op.cc61 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
64 context, data_format_ == FORMAT_NHWC, in AvgPoolingOp()
94 data_format_, in Compute()
118 TensorFormat data_format_; member in tensorflow::AvgPoolingOp
139 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
150 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingOp()
151 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingOp()
164 data_format_, in Compute()
188 data_format_, tensor_in, output_shape, in Compute()
191 if (data_format_ == FORMAT_NCHW) { in Compute()
[all …]
Dspacetodepth_op.cc64 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
74 context, data_format_ == FORMAT_NHWC, in SpaceToDepthOp()
84 const bool is_int8x4 = (data_format_ == FORMAT_NCHW_VECT_C); in Compute()
98 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); in Compute()
100 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); in Compute()
102 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); in Compute()
104 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * in Compute()
125 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
131 if (data_format_ == FORMAT_NCHW_VECT_C) { in Compute()
138 } else if (data_format_ == FORMAT_NCHW) { in Compute()
[all …]
Ddepthtospace_op.cc49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
59 context, data_format_ == FORMAT_NHWC, in DepthToSpaceOp()
71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
83 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'N')); in Compute()
85 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'H')); in Compute()
87 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'W')); in Compute()
89 input.dim_size(GetTensorDimIndex<kNumSpatialDims>(data_format_, 'C')) * in Compute()
109 ShapeFromFormat(data_format_, batch_size, output_height, in Compute()
124 } else if (data_format_ == FORMAT_NCHW) { in Compute()
142 TensorFormat data_format_; member in tensorflow::DepthToSpaceOp
Ddepthwise_conv_op.cc276 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeOp()
282 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeOp()
283 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp()
284 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp()
285 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp()
299 /*num_dims=*/4, data_format_)); in DepthwiseConv2dNativeOp()
312 (data_format_ == FORMAT_NCHW || in DepthwiseConv2dNativeOp()
313 (data_format_ == FORMAT_NHWC && stride_ == stride_w && in DepthwiseConv2dNativeOp()
319 use_cudnn_grouped_conv_ = dtype_ == DT_HALF && data_format_ == FORMAT_NCHW; in DepthwiseConv2dNativeOp()
343 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute()
[all …]
Dmaxpooling_op.cc226 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
229 context, data_format_ == FORMAT_NHWC, in MaxPoolingGradOp()
258 /*num_dims=*/4, data_format_)); in MaxPoolingGradOp()
339 TensorFormat data_format_; member in tensorflow::MaxPoolingGradOp
352 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
363 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradOp()
364 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradOp()
374 /*num_dims=*/4, data_format_)); in MaxPoolingGradOp()
415 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N'); in Compute()
416 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute()
[all …]
Dconv_ops_3d.cc85 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DOp()
93 (GetTensorDim(stride_, data_format_, 'N') == 1 && in Conv3DOp()
94 GetTensorDim(stride_, data_format_, 'C') == 1), in Conv3DOp()
99 (GetTensorDim(stride_, data_format_, '0') > 0 && in Conv3DOp()
100 GetTensorDim(stride_, data_format_, '1') > 0 && in Conv3DOp()
101 GetTensorDim(stride_, data_format_, '2') > 0), in Conv3DOp()
108 (GetTensorDim(dilation_, data_format_, 'N') == 1 && in Conv3DOp()
109 GetTensorDim(dilation_, data_format_, 'C') == 1), in Conv3DOp()
115 (GetTensorDim(dilation_, data_format_, '0') > 0 && in Conv3DOp()
116 GetTensorDim(dilation_, data_format_, '1') > 0 && in Conv3DOp()
[all …]
Dconv_grad_input_ops.h359 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
366 int stride_n = GetTensorDim(strides_, data_format_, 'N');
367 int stride_c = GetTensorDim(strides_, data_format_, 'C');
368 int stride_h = GetTensorDim(strides_, data_format_, 'H');
369 int stride_w = GetTensorDim(strides_, data_format_, 'W');
382 int dilation_n = GetTensorDim(dilations_, data_format_, 'N');
383 int dilation_c = GetTensorDim(dilations_, data_format_, 'C');
384 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
385 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
398 /*num_dims=*/4, data_format_));
[all …]
Dconv_ops_using_gemm.cc438 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DUsingGemmOp()
440 OP_REQUIRES(context, data_format_ == FORMAT_NHWC, in Conv2DUsingGemmOp()
446 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp()
447 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp()
481 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute()
492 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute()
502 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); in Compute()
511 const int64 batch_raw = GetTensorDim(input, data_format_, 'N'); in Compute()
519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
[all …]
Ddepthwise_conv_grad_op.cc75 const int64 input_rows_raw = GetTensorDim(input_shape, data_format_, 'H'); \
81 const int64 input_cols_raw = GetTensorDim(input_shape, data_format_, 'W'); \
90 GetTensorDim(out_backprop.shape(), data_format_, 'H'); \
97 GetTensorDim(out_backprop.shape(), data_format_, 'W'); \
103 const int64 in_depth = GetTensorDim(input_shape, data_format_, 'C'); \
109 GetTensorDim(out_backprop.shape(), data_format_, 'C'); \
123 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', &pad_top, \
125 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'W', &pad_left, \
560 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeBackpropInputOp()
563 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropInputOp()
[all …]
Dconv_grad_filter_ops.cc267 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DBackpropFilterOp()
270 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DBackpropFilterOp()
271 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DBackpropFilterOp()
272 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropFilterOp()
273 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DBackpropFilterOp()
285 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DBackpropFilterOp()
286 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DBackpropFilterOp()
287 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp()
288 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp()
301 /*num_dims=*/4, data_format_)); in Conv2DBackpropFilterOp()
[all …]
Dconv_ops_gpu.h112 data_format_(data_format), in batch_()
148 ::tensorflow::ToString(data_format_), ", ", in ToString()
183 return std::make_tuple(batch_, in_depths_, in_, data_format_, out_depths_, in get_data_as_tuple()
215 TensorFormat data_format_; variable
Dpooling_ops_common.h101 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingOp()
104 context, data_format_ == FORMAT_NHWC, in MaxPoolingOp()
109 data_format_ = FORMAT_NHWC; in MaxPoolingOp()
300 TensorFormat data_format_; variable
340 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
344 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
349 data_format_ = FORMAT_NHWC;
401 data_format_,
560 TensorFormat data_format_;
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc34 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasOp()
37 data_format_ = FORMAT_NHWC; in BiasOp()
53 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile()
72 TensorFormat data_format_; member in tensorflow::__anonf08d60ad0111::BiasOp
83 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasAddGradOp()
86 data_format_ = FORMAT_NHWC; in BiasAddGradOp()
98 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile()
122 TensorFormat data_format_; member in tensorflow::__anonf08d60ad0111::BiasAddGradOp
Ddepthtospace_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
43 TensorFormat data_format = data_format_; in Compile()
55 ToString(data_format_))); in Compile()
170 if (data_format != data_format_) { in Compile()
171 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
181 TensorFormat data_format_; member in tensorflow::__anon4abce69f0111::DepthToSpaceOp
Dspacetodepth_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
43 TensorFormat data_format = data_format_; in Compile()
55 ToString(data_format_))); in Compile()
174 if (data_format != data_format_) { in Compile()
175 DCHECK(data_format == FORMAT_NCHW && data_format_ == FORMAT_NCHW_VECT_C); in Compile()
185 TensorFormat data_format_; member in tensorflow::__anonba7f0abf0111::SpaceToDepthOp
Dpooling_ops.cc130 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anon9297e9860111::PoolingOp
159 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp()
163 data_format_ != FORMAT_NCHW_VECT_C && in MaxPoolOp()
164 data_format_ != FORMAT_NHWC_VECT_W, in MaxPoolOp()
187 XlaTensorFormat(data_format_, input_shape.dims() - 2)); in Compile()
218 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
238 XlaTensorFormat(data_format_, input_shape.dims() - 2); in Compile()
344 XlaTensorFormat(data_format_, tensor_in_shape.dims() - 2)); in Compile()
370 TensorFormat data_format_ = FORMAT_NHWC; member in tensorflow::__anon9297e9860111::MaxPoolGradOp
379 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradOp()
[all …]
Dbatch_norm_op.cc45 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
86 GetTensorFeatureDimIndex(input_shape.dims(), data_format_); in CompileImpl()
197 TensorFormat data_format_; member in tensorflow::__anon73cf47b80111::FusedBatchNormOp
246 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
269 GetTensorFeatureDimIndex(input_dims, data_format_); in Compile()
344 TensorFormat data_format_; member in tensorflow::__anon73cf47b80111::FusedBatchNormGradOp

12