Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d.cc58 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Pool3dParameters()
59 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters()
60 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters()
61 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Pool3dParameters()
62 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Pool3dParameters()
63 window_planes = GetTensorDim(ksize, data_format, '0'); in Pool3dParameters()
64 window_rows = GetTensorDim(ksize, data_format, '1'); in Pool3dParameters()
65 window_cols = GetTensorDim(ksize, data_format, '2'); in Pool3dParameters()
66 depth_window = GetTensorDim(ksize, data_format, 'C'); in Pool3dParameters()
67 plane_stride = GetTensorDim(stride, data_format, '0'); in Pool3dParameters()
[all …]
Dconv_ops_3d.cc93 (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()
117 GetTensorDim(dilation_, data_format_, '2') > 0), in Conv3DOp()
[all …]
Dconv_grad_ops_3d.cc197 (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()
304 (GetTensorDim(dilation_, data_format_, 'C') == 1 && in Conv3DCustomBackpropInputOp()
305 GetTensorDim(dilation_, data_format_, 'N') == 1), in Conv3DCustomBackpropInputOp()
312 (GetTensorDim(dilation_, data_format_, '0') == 1 && in Conv3DCustomBackpropInputOp()
[all …]
Dconv_ops.cc158 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in operator ()()
196 const int64 in_depth = GetTensorDim(input, data_format, 'C'); in operator ()()
371 const int64 stride_n = GetTensorDim(strides, data_format, 'N'); in InitConv2DParameters()
372 const int64 stride_c = GetTensorDim(strides, data_format, 'C'); in InitConv2DParameters()
373 const int64 stride_h = GetTensorDim(strides, data_format, 'H'); in InitConv2DParameters()
374 const int64 stride_w = GetTensorDim(strides, data_format, 'W'); in InitConv2DParameters()
383 const int64 dilation_n = GetTensorDim(dilations, data_format, 'N'); in InitConv2DParameters()
384 const int64 dilation_c = GetTensorDim(dilations, data_format, 'C'); in InitConv2DParameters()
385 const int64 dilation_h = GetTensorDim(dilations, data_format, 'H'); in InitConv2DParameters()
386 const int64 dilation_w = GetTensorDim(dilations, data_format, 'W'); in InitConv2DParameters()
[all …]
Dcudnn_pooling_gpu.cc44 const int64 in_batch = GetTensorDim(tensor_in, data_format, 'N'); in Compute()
45 const int64 in_features = GetTensorDim(tensor_in, data_format, 'C'); in Compute()
87 GetTensorDim(tensor_in, data_format, '2' - i)); in Compute()
89 GetTensorDim(out_shape, data_format, '2' - i)); in Compute()
145 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Compute()
146 const int64 in_features = GetTensorDim(tensor_in_shape, data_format, 'C'); in Compute()
223 dim_i, GetTensorDim(tensor_in_shape, data_format, '2' - i)); in Compute()
Dconv_ops_using_gemm.cc446 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()
Dfused_batch_norm_op.cc122 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
123 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
124 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()()
125 const int64 in_depths = GetTensorDim(x_input, tensor_format, 'C'); in operator ()()
260 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
261 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
262 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()()
263 const int64 in_depths = GetTensorDim(x_input, tensor_format, 'C'); in operator ()()
352 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
353 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
[all …]
Dconv_grad_filter_ops.cc270 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()
349 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
350 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
[all …]
Dconv_grad_input_ops.h366 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');
441 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
442 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
[all …]
Ddepthwise_conv_op.cc282 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()
343 const int64 in_depth = GetTensorDim(input, data_format_, 'C'); in Compute()
355 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H'); in Compute()
363 const int64 input_cols_raw = GetTensorDim(input, data_format_, 'W'); in Compute()
Dconv_grad_input_ops.cc251 .set_height(GetTensorDim(compatible_input_shape, data_format, 'H')) in operator ()()
252 .set_width(GetTensorDim(compatible_input_shape, data_format, 'W')) in operator ()()
340 GetTensorDim(compatible_input_shape, data_format, 'N'), in operator ()()
341 GetTensorDim(compatible_input_shape, data_format, 'H'), in operator ()()
342 GetTensorDim(compatible_input_shape, data_format, 'W'), in operator ()()
343 GetTensorDim(compatible_input_shape, data_format, 'C')), in operator ()()
516 GetTensorDim(input_shape, data_format, 'N'), in operator ()()
517 GetTensorDim(input_shape, data_format, 'H'), in operator ()()
518 GetTensorDim(input_shape, data_format, 'W'), in operator ()()
519 GetTensorDim(input_shape, data_format, 'C')), in operator ()()
Dpooling_ops_common.cc133 depth = GetTensorDim(tensor_in_shape, data_format, 'C') * in PoolParameters()
135 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters()
136 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
137 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in PoolParameters()
138 window_rows = GetTensorDim(ksize, data_format, 'H'); in PoolParameters()
139 window_cols = GetTensorDim(ksize, data_format, 'W'); in PoolParameters()
140 depth_window = GetTensorDim(ksize, data_format, 'C'); in PoolParameters()
141 row_stride = GetTensorDim(stride, data_format, 'H'); in PoolParameters()
142 col_stride = GetTensorDim(stride, data_format, 'W'); in PoolParameters()
143 depth_stride = GetTensorDim(stride, data_format, 'C'); in PoolParameters()
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'); \
563 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropInputOp()
564 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp()
565 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp()
566 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp()
[all …]
Davgpooling_op.cc150 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingOp()
151 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingOp()
431 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingGradOp()
432 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOp()
518 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel()
519 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in AvgPoolingGradOpCustomGPUKernel()
Dconv_ops_fused_impl.h460 const int64 in_batch = GetTensorDim(input, params.data_format, 'N');
461 int64 in_rows = GetTensorDim(input, params.data_format, 'H');
462 int64 in_cols = GetTensorDim(input, params.data_format, 'W');
463 const int64 in_depths = GetTensorDim(input, params.data_format, 'C');
469 const int64 out_batch = GetTensorDim(*output, params.data_format, 'N');
470 const int64 out_rows = GetTensorDim(*output, params.data_format, 'H');
471 const int64 out_cols = GetTensorDim(*output, params.data_format, 'W');
472 const int64 out_depths = GetTensorDim(*output, params.data_format, 'C');
Dmaxpooling_op.cc363 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradOp()
364 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); 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()
680 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradGradOp()
681 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N'); in MaxPoolingGradGradOp()
728 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N'); in Compute()
729 const int32 stride_n = GetTensorDim(stride, data_format_, 'N'); in Compute()
999 GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
1030 const int64 batch_size = GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
[all …]
Dconv_ops_fused_image_transform.cc637 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp()
638 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp()
819 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute()
820 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute()
Dconv_grad_shape_utils.cc185 const int batch_size = GetTensorDim(out_backprop_shape, data_format, 'N'); in Conv2DBackpropComputeInputShape()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h82 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()
104 int dilations_rows = GetTensorDim(dilations_, data_format_, '1'); in GetDilationsInMklOrder()
105 int dilations_cols = GetTensorDim(dilations_, data_format_, '2'); in GetDilationsInMklOrder()
[all …]
Dmkl_pooling_ops_common.cc235 depth = GetTensorDim(tensor_in_shape, data_format, 'C'); in Init()
238 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in Init()
239 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init()
242 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init()
243 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init()
244 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Init()
246 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); in Init()
288 window_rows = GetTensorDim(ksize, data_format, 'H'); in Init()
289 window_cols = GetTensorDim(ksize, data_format, 'W'); in Init()
290 depth_window = GetTensorDim(ksize, data_format, 'C'); in Init()
[all …]
Dmkl_conv_ops.cc449 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()
491 GetTensorDim(dilations_, data_format_, '1') > 0 && in MklConvOp()
[all …]
Dmkl_fused_batch_norm_op.cc930 depth_ = static_cast<int>(GetTensorDim(input, tensor_format_, 'C')); in ExtractParams()
1313 depth_ = static_cast<int>(GetTensorDim(input, tensor_format_, 'C')); in ExtractParams()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h436 T GetTensorDim(gtl::ArraySlice<T> dimension_attributes, in GetTensorDim() function
461 T GetTensorDim(const std::vector<T>& attributes, TensorFormat format, in GetTensorDim() function
463 return GetTensorDim(gtl::ArraySlice<T>(attributes), format, dimension); in GetTensorDim()
468 inline int64 GetTensorDim(const TensorShape& tensor_shape, in GetTensorDim() function
470 return GetTensorDim(gtl::ArraySlice<int64>(tensor_shape.dim_sizes()), in GetTensorDim()
485 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format, in GetTensorDim() function
487 return GetTensorDim(tensor.shape(), tensor_format, dimension); in GetTensorDim()
597 const int64 batch = GetTensorDim(src_shape, src_format, 'N'); in ShapeFromFormat()
598 const int64 channels = GetTensorDim(src_shape, src_format, 'C') * in ShapeFromFormat()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc631 const int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in Conv2DShapeImpl()
632 const int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in Conv2DShapeImpl()
633 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H'); in Conv2DShapeImpl()
634 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W'); in Conv2DShapeImpl()
1087 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in AvgPoolShape()
1088 int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in AvgPoolShape()
1089 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in AvgPoolShape()
1090 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape()
1528 int32 stride_depth = GetTensorDim(strides, data_format, 'C'); in MaxPoolShapeImpl()
1529 int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in MaxPoolShapeImpl()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1578 if (GetTensorDim(ksize, data_format, 'N') == 1 && in NonDepthBatchWisePoolRewrite()
1579 GetTensorDim(strides, data_format, 'N') == 1 && in NonDepthBatchWisePoolRewrite()
1580 GetTensorDim(ksize, data_format, 'C') == 1 && in NonDepthBatchWisePoolRewrite()
1581 GetTensorDim(strides, data_format, 'C') == 1) { in NonDepthBatchWisePoolRewrite()