| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.cc | 37 depth = GetTensorDim(tensor_in_shape, data_format, 'C');
38 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W');
39 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H');
40 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N');
85 window_rows = GetTensorDim(ksize, data_format, 'H');
86 window_cols = GetTensorDim(ksize, data_format, 'W');
87 depth_window = GetTensorDim(ksize, data_format, 'C');
90 row_stride = GetTensorDim(stride, data_format, 'H');
91 col_stride = GetTensorDim(stride, data_format, 'W');
92 depth_stride = GetTensorDim(stride, data_format, 'C')
[all...] |
| pooling_ops_3d.cc | 63 depth = GetTensorDim(tensor_in_shape, data_format, 'C');
64 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0');
65 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1');
66 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2');
67 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N');
68 window_planes = GetTensorDim(ksize, data_format, '0');
69 window_rows = GetTensorDim(ksize, data_format, '1');
70 window_cols = GetTensorDim(ksize, data_format, '2');
71 depth_window = GetTensorDim(ksize, data_format, 'C');
72 plane_stride = GetTensorDim(stride, data_format, '0')
[all...] |
| conv_ops_3d.cc | 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'),
120 GetTensorDim(stride_, data_format_, '2')}}
[all...] |
| conv_ops.cc | 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');
339 const int64 input_rows_raw = GetTensorDim(input, data_format_, 'H')
[all...] |
| pooling_ops_common.cc | 46 depth = GetTensorDim(tensor_in_shape, data_format, 'C') *
48 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W');
49 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H');
50 tensor_in_batch = GetTensorDim(tensor_in_shape, data_format, 'N');
51 window_rows = GetTensorDim(ksize, data_format, 'H');
52 window_cols = GetTensorDim(ksize, data_format, 'W');
53 depth_window = GetTensorDim(ksize, data_format, 'C');
54 row_stride = GetTensorDim(stride, data_format, 'H');
55 col_stride = GetTensorDim(stride, data_format, 'W');
56 depth_stride = GetTensorDim(stride, data_format, 'C')
[all...] |
| cudnn_pooling_gpu.cc | 43 const int64 in_batch = GetTensorDim(tensor_in, data_format, 'N');
44 const int64 in_features = GetTensorDim(tensor_in, data_format, 'C');
86 GetTensorDim(tensor_in, data_format, '2' - i));
88 GetTensorDim(out_shape, data_format, '2' - i));
130 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N');
131 const int64 in_features = GetTensorDim(tensor_in_shape, data_format, 'C');
208 dim_i, GetTensorDim(tensor_in_shape, data_format, '2' - i));
|
| mkl_conv_grad_bias_ops.cc | 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');
|
| pooling_ops_3d_sycl.h | 186 const int out_planes = GetTensorDim(*output, data_format, '0');
187 const int out_rows = GetTensorDim(*output, data_format, '1');
188 const int out_cols = GetTensorDim(*output, data_format, '2');
189 const int batch = GetTensorDim(tensor_in, data_format, 'N');
190 const int in_planes = GetTensorDim(tensor_in, data_format, '0');
191 const int in_rows = GetTensorDim(tensor_in, data_format, '1');
192 const int in_cols = GetTensorDim(tensor_in, data_format, '2');
193 const int depth = GetTensorDim(tensor_in, data_format, 'C');
355 const int batch = GetTensorDim(tensor_in, data_format, 'N');
356 const int in_planes = GetTensorDim(tensor_in, data_format, '0')
[all...] |
| mkl_conv_ops.h | 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');
225 int input_cols = GetTensorDim(input_shape, data_format_, 'W');
237 int out_batch = GetTensorDim(input_shape, data_format_, 'N')
[all...] |
| conv_ops_using_gemm.cc | 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');
|
| conv_grad_input_ops.cc | 621 int stride_n = GetTensorDim(strides_, data_format_, 'N');
622 int stride_c = GetTensorDim(strides_, data_format_, 'C');
623 int stride_h = GetTensorDim(strides_, data_format_, 'H');
624 int stride_w = GetTensorDim(strides_, data_format_, 'W');
636 int dilation_n = GetTensorDim(dilations_, data_format_, 'N');
637 int dilation_c = GetTensorDim(dilations_, data_format_, 'C');
638 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
639 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
677 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
678 const int stride_cols = GetTensorDim(strides_, data_format_, 'W')
[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'),
[all...] |
| depthwise_conv_op.cc | 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');
|
| conv_grad_filter_ops.cc | 546 int stride_n = GetTensorDim(strides_, data_format_, 'N');
547 int stride_c = GetTensorDim(strides_, data_format_, 'C');
548 int stride_h = GetTensorDim(strides_, data_format_, 'H');
549 int stride_w = GetTensorDim(strides_, data_format_, 'W');
561 int dilation_n = GetTensorDim(dilations_, data_format_, 'N');
562 int dilation_c = GetTensorDim(dilations_, data_format_, 'C');
563 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
564 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
608 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
609 const int stride_cols = GetTensorDim(strides_, data_format_, 'W')
[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'); \
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');
540 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C')
[all...] |
| mkl_conv_ops.cc | 80 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
81 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
127 : GetTensorDim(input, data_format_, 'C');
139 : GetTensorDim(input, data_format_, 'H');
151 : GetTensorDim(input, data_format_, 'W');
162 : GetTensorDim(input, data_format_, 'N');
171 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
172 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
503 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
504 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C')
[all...] |
| avgpooling_op.cc | 131 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
132 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N');
387 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
388 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N');
461 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
462 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N');
|
| fused_batch_norm_op.cc | 231 const int64 batch_size = GetTensorDim(x, tensor_format, 'N');
232 const int64 channels = GetTensorDim(x, tensor_format, 'C');
233 const int64 height = GetTensorDim(x, tensor_format, 'H');
234 const int64 width = GetTensorDim(x, tensor_format, 'W');
372 const int64 batch_size = GetTensorDim(x, tensor_format, 'N');
373 const int64 channels = GetTensorDim(x, tensor_format, 'C');
374 const int64 height = GetTensorDim(x, tensor_format, 'H');
375 const int64 width = GetTensorDim(x, tensor_format, 'W');
[all...] |
| maxpooling_op.cc | 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');
658 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N');
659 const int32 stride_n = GetTensorDim(stride_, data_format_, 'N');
706 const int32 ksize_n = GetTensorDim(ksize, data_format_, 'N');
707 const int32 stride_n = GetTensorDim(stride, data_format_, 'N');
[all...] |
| mkl_fused_batch_norm_op.cc | 249 : GetTensorDim(input, tensor_format, 'W'));
252 : GetTensorDim(input, tensor_format, 'H'));
255 : GetTensorDim(input, tensor_format, 'C'));
258 : GetTensorDim(input, tensor_format, 'N'));
532 : GetTensorDim(input, tensor_format, 'W'));
535 : GetTensorDim(input, tensor_format, 'H'));
538 : GetTensorDim(input, tensor_format, 'C'));
541 : GetTensorDim(input, tensor_format, 'N'));
[all...] |
| conv_ops_fused.cc | 635 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N');
636 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C');
[all...] |
| mkl_conv_grad_filter_ops.cc | 71 int stride_n = GetTensorDim(strides_, data_format_, 'N');
72 int stride_c = GetTensorDim(strides_, data_format_, 'C');
|
| /external/tensorflow/tensorflow/core/util/ |
| tensor_format.h | 334 T GetTensorDim(gtl::ArraySlice<T> dimension_attributes,
362 T GetTensorDim(const std::vector<T>& attributes, TensorFormat format,
364 return GetTensorDim(gtl::ArraySlice<T>(attributes), format, dimension);
369 inline int64 GetTensorDim(const TensorShape& tensor_shape,
371 return GetTensorDim(gtl::ArraySlice<int64>(tensor_shape.dim_sizes()),
386 inline int64 GetTensorDim(const Tensor& tensor, TensorFormat tensor_format,
388 return GetTensorDim(tensor.shape(), tensor_format, dimension);
478 const int64 batch = GetTensorDim(src_shape, src_format, 'N');
479 const int64 channels = GetTensorDim(src_shape, src_format, 'C') *
484 {{GetTensorDim(src_shape, src_format, '0')
[all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 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),
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');
335 const int batch_size = GetTensorDim(conv_input_param, data_format, 'N');
336 int conv_input_rows = GetTensorDim(conv_input_param, data_format, 'H');
337 int conv_input_cols = GetTensorDim(conv_input_param, data_format, 'W')
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 424 const int32 stride_rows = GetTensorDim(strides, data_format, 'H');
425 const int32 stride_cols = GetTensorDim(strides, data_format, 'W');
426 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H');
427 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W');
660 int32 stride_rows = GetTensorDim(strides, data_format, 'H');
661 int32 stride_cols = GetTensorDim(strides, data_format, 'W');
662 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H');
663 int32 kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W');
810 int32 stride_depth = GetTensorDim(strides, data_format, 'C');
811 int32 stride_rows = GetTensorDim(strides, data_format, 'H')
[all...] |
