| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 30 int64 input_cols, int64 input_channels, int64 kernel_rows, in EigenConvImpl() argument
44 kernel(rhs, kernel_rows, kernel_cols, kernel_channels, kernel_filters); in EigenConvImpl()
59 pre_contract_dims[1] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
69 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
75 .extract_image_patches(kernel_cols, kernel_rows, col_stride, in EigenConvImpl()
|
| D | runtime_single_threaded_conv2d.cc | 28 int64 input_channels, int64 kernel_rows, int64 kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
46 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 30 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_EigenConvF32() argument
39 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF32()
48 int64 input_channels, int64 kernel_rows, int64 kernel_cols, in __xla_cpu_runtime_EigenConvF16() argument
58 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 54 int64 input_cols, int64 input_channels, int64 kernel_rows, in MKLConvImpl() argument
71 ToInt(kernel_channels), ToInt(kernel_rows), in MKLConvImpl()
155 int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, in __xla_cpu_runtime_MKLConvF32() argument
166 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
172 input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_single_threaded_conv2d.h | 29 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols,
42 tensorflow::int64 input_channels, tensorflow::int64 kernel_rows,
|
| D | runtime_conv2d.h | 28 tensorflow::int64 input_channels, tensorflow::int64 kernel_rows,
43 tensorflow::int64 kernel_rows, tensorflow::int64 kernel_cols,
|
| D | runtime_conv2d_mkl.h | 28 tensorflow::int64 input_channels, tensorflow::int64 kernel_rows,
|
| D | ir_emitter.cc | 1137 int64 kernel_rows = in HandleConvolution() local
1225 b_.getInt64(kernel_rows), in HandleConvolution()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 813 int kernel_rows, int kernel_cols, int stride, in BM_AvgPool() argument
832 CHECK_EQ(kernel_rows, kernel_cols); in BM_AvgPool()
835 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_AvgPool()
904 int depth, int kernel_rows, int kernel_cols, in BM_AvgPoolBk() argument
919 TF_CHECK_OK(GetWindowedOutputSize(rows, kernel_rows, stride, padding, in BM_AvgPoolBk()
941 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_AvgPoolBk()
1006 int kernel_rows, int kernel_cols, int stride, in BM_MaxPool() argument
1028 CHECK_EQ(kernel_rows, kernel_cols); in BM_MaxPool()
1031 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in BM_MaxPool()
1101 int depth, int kernel_rows, int kernel_cols, in BM_MaxPoolBk() argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 755 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in AvgPoolShape() local
777 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in AvgPoolShape()
907 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in MaxPoolShape() local
926 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in MaxPoolShape()
1009 int32 kernel_rows = GetTensorDim(kernel_sizes, data_format, 'H'); in MaxPoolV2Shape() local
1028 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in MaxPoolV2Shape()
1067 int32 kernel_planes, kernel_rows, kernel_cols; in Pool3DShape() local
1080 kernel_rows = kernel_sizes[3]; in Pool3DShape()
1087 kernel_rows = kernel_sizes[2]; in Pool3DShape()
1107 c, in_rows_dim, kernel_rows, stride_rows, padding, &output_rows)); in Pool3DShape()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1110 int kernel_rows = in TEST_F() local
1121 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in TEST_F()
2314 int kernel_rows = in TEST_F() local
2326 .Attr("ksize", {1, kernel_rows, kernel_cols, 1}) in TEST_F()
|