| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_image_patch.cpp | 21 VERIFY_IS_EQUAL(tensor.dimension(0), tensor_row_major.dimension(3)); in test_simple_patch()
22 VERIFY_IS_EQUAL(tensor.dimension(1), tensor_row_major.dimension(2)); in test_simple_patch()
23 VERIFY_IS_EQUAL(tensor.dimension(2), tensor_row_major.dimension(1)); in test_simple_patch()
24 VERIFY_IS_EQUAL(tensor.dimension(3), tensor_row_major.dimension(0)); in test_simple_patch()
29 VERIFY_IS_EQUAL(single_pixel_patch.dimension(0), 2); in test_simple_patch()
30 VERIFY_IS_EQUAL(single_pixel_patch.dimension(1), 1); in test_simple_patch()
31 VERIFY_IS_EQUAL(single_pixel_patch.dimension(2), 1); in test_simple_patch()
32 VERIFY_IS_EQUAL(single_pixel_patch.dimension(3), 3*5); in test_simple_patch()
33 VERIFY_IS_EQUAL(single_pixel_patch.dimension(4), 7); in test_simple_patch()
38 VERIFY_IS_EQUAL(single_pixel_patch_row_major.dimension(0), 7); in test_simple_patch()
[all …]
|
| D | cxx11_tensor_patch.cpp | 32 VERIFY_IS_EQUAL(no_patch.dimension(0), 1); in test_simple_patch()
33 VERIFY_IS_EQUAL(no_patch.dimension(1), 1); in test_simple_patch()
34 VERIFY_IS_EQUAL(no_patch.dimension(2), 1); in test_simple_patch()
35 VERIFY_IS_EQUAL(no_patch.dimension(3), 1); in test_simple_patch()
36 VERIFY_IS_EQUAL(no_patch.dimension(4), tensor.size()); in test_simple_patch()
38 VERIFY_IS_EQUAL(no_patch.dimension(0), tensor.size()); in test_simple_patch()
39 VERIFY_IS_EQUAL(no_patch.dimension(1), 1); in test_simple_patch()
40 VERIFY_IS_EQUAL(no_patch.dimension(2), 1); in test_simple_patch()
41 VERIFY_IS_EQUAL(no_patch.dimension(3), 1); in test_simple_patch()
42 VERIFY_IS_EQUAL(no_patch.dimension(4), 1); in test_simple_patch()
[all …]
|
| D | cxx11_tensor_volume_patch.cpp | 15 VERIFY_IS_EQUAL(single_voxel_patch.dimension(0), 4); in test_single_voxel_patch()
16 VERIFY_IS_EQUAL(single_voxel_patch.dimension(1), 1); in test_single_voxel_patch()
17 VERIFY_IS_EQUAL(single_voxel_patch.dimension(2), 1); in test_single_voxel_patch()
18 VERIFY_IS_EQUAL(single_voxel_patch.dimension(3), 1); in test_single_voxel_patch()
19 VERIFY_IS_EQUAL(single_voxel_patch.dimension(4), 2 * 3 * 5); in test_single_voxel_patch()
20 VERIFY_IS_EQUAL(single_voxel_patch.dimension(5), 7); in test_single_voxel_patch()
24 VERIFY_IS_EQUAL(single_voxel_patch_row_major.dimension(0), 7); in test_single_voxel_patch()
25 VERIFY_IS_EQUAL(single_voxel_patch_row_major.dimension(1), 2 * 3 * 5); in test_single_voxel_patch()
26 VERIFY_IS_EQUAL(single_voxel_patch_row_major.dimension(2), 1); in test_single_voxel_patch()
27 VERIFY_IS_EQUAL(single_voxel_patch_row_major.dimension(3), 1); in test_single_voxel_patch()
[all …]
|
| D | cxx11_tensor_reverse.cpp | 33 VERIFY_IS_EQUAL(reversed_tensor.dimension(0), 2); in test_simple_reverse()
34 VERIFY_IS_EQUAL(reversed_tensor.dimension(1), 3); in test_simple_reverse()
35 VERIFY_IS_EQUAL(reversed_tensor.dimension(2), 5); in test_simple_reverse()
36 VERIFY_IS_EQUAL(reversed_tensor.dimension(3), 7); in test_simple_reverse()
55 VERIFY_IS_EQUAL(reversed_tensor.dimension(0), 2); in test_simple_reverse()
56 VERIFY_IS_EQUAL(reversed_tensor.dimension(1), 3); in test_simple_reverse()
57 VERIFY_IS_EQUAL(reversed_tensor.dimension(2), 5); in test_simple_reverse()
58 VERIFY_IS_EQUAL(reversed_tensor.dimension(3), 7); in test_simple_reverse()
78 VERIFY_IS_EQUAL(reversed_tensor.dimension(0), 2); in test_simple_reverse()
79 VERIFY_IS_EQUAL(reversed_tensor.dimension(1), 3); in test_simple_reverse()
[all …]
|
| D | cxx11_tensor_shuffling.cpp | 31 VERIFY_IS_EQUAL(no_shuffle.dimension(0), 2); in test_simple_shuffling()
32 VERIFY_IS_EQUAL(no_shuffle.dimension(1), 3); in test_simple_shuffling()
33 VERIFY_IS_EQUAL(no_shuffle.dimension(2), 5); in test_simple_shuffling()
34 VERIFY_IS_EQUAL(no_shuffle.dimension(3), 7); in test_simple_shuffling()
53 VERIFY_IS_EQUAL(shuffle.dimension(0), 5); in test_simple_shuffling()
54 VERIFY_IS_EQUAL(shuffle.dimension(1), 7); in test_simple_shuffling()
55 VERIFY_IS_EQUAL(shuffle.dimension(2), 3); in test_simple_shuffling()
56 VERIFY_IS_EQUAL(shuffle.dimension(3), 2); in test_simple_shuffling()
98 VERIFY_IS_EQUAL(result.dimension(0), 5); in test_expr_shuffling()
99 VERIFY_IS_EQUAL(result.dimension(1), 7); in test_expr_shuffling()
[all …]
|
| D | cxx11_tensor_layout_swap.cpp | 22 VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2)); in test_simple_swap()
23 VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1)); in test_simple_swap()
24 VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0)); in test_simple_swap()
43 VERIFY_IS_EQUAL(tensor.dimension(0), tensor2.dimension(2)); in test_swap_as_lvalue()
44 VERIFY_IS_EQUAL(tensor.dimension(1), tensor2.dimension(1)); in test_swap_as_lvalue()
45 VERIFY_IS_EQUAL(tensor.dimension(2), tensor2.dimension(0)); in test_swap_as_lvalue()
|
| D | cxx11_tensor_broadcasting.cpp | 30 VERIFY_IS_EQUAL(no_broadcast.dimension(0), 2); in test_simple_broadcasting()
31 VERIFY_IS_EQUAL(no_broadcast.dimension(1), 3); in test_simple_broadcasting()
32 VERIFY_IS_EQUAL(no_broadcast.dimension(2), 5); in test_simple_broadcasting()
33 VERIFY_IS_EQUAL(no_broadcast.dimension(3), 7); in test_simple_broadcasting()
52 VERIFY_IS_EQUAL(broadcast.dimension(0), 4); in test_simple_broadcasting()
53 VERIFY_IS_EQUAL(broadcast.dimension(1), 9); in test_simple_broadcasting()
54 VERIFY_IS_EQUAL(broadcast.dimension(2), 5); in test_simple_broadcasting()
55 VERIFY_IS_EQUAL(broadcast.dimension(3), 28); in test_simple_broadcasting()
82 VERIFY_IS_EQUAL(broadcast.dimension(0), 16); in test_vectorized_broadcasting()
83 VERIFY_IS_EQUAL(broadcast.dimension(1), 9); in test_vectorized_broadcasting()
[all …]
|
| D | cxx11_tensor_ifft.cpp | 31 VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), sequence_length); in test_1D_fft_ifft_invariant()
32 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), sequence_length); in test_1D_fft_ifft_invariant()
54 VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), dim0); in test_2D_fft_ifft_invariant()
55 VERIFY_IS_EQUAL(tensor_after_fft.dimension(1), dim1); in test_2D_fft_ifft_invariant()
56 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), dim0); in test_2D_fft_ifft_invariant()
57 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(1), dim1); in test_2D_fft_ifft_invariant()
83 VERIFY_IS_EQUAL(tensor_after_fft.dimension(0), dim0); in test_3D_fft_ifft_invariant()
84 VERIFY_IS_EQUAL(tensor_after_fft.dimension(1), dim1); in test_3D_fft_ifft_invariant()
85 VERIFY_IS_EQUAL(tensor_after_fft.dimension(2), dim2); in test_3D_fft_ifft_invariant()
86 VERIFY_IS_EQUAL(tensor_after_fft_ifft.dimension(0), dim0); in test_3D_fft_ifft_invariant()
[all …]
|
| D | cxx11_tensor_chipping.cpp | 25 VERIFY_IS_EQUAL(chip1.dimension(0), 3); in test_simple_chip()
26 VERIFY_IS_EQUAL(chip1.dimension(1), 5); in test_simple_chip()
27 VERIFY_IS_EQUAL(chip1.dimension(2), 7); in test_simple_chip()
28 VERIFY_IS_EQUAL(chip1.dimension(3), 11); in test_simple_chip()
41 VERIFY_IS_EQUAL(chip2.dimension(0), 2); in test_simple_chip()
42 VERIFY_IS_EQUAL(chip2.dimension(1), 5); in test_simple_chip()
43 VERIFY_IS_EQUAL(chip2.dimension(2), 7); in test_simple_chip()
44 VERIFY_IS_EQUAL(chip2.dimension(3), 11); in test_simple_chip()
56 VERIFY_IS_EQUAL(chip3.dimension(0), 2); in test_simple_chip()
57 VERIFY_IS_EQUAL(chip3.dimension(1), 3); in test_simple_chip()
[all …]
|
| D | cxx11_tensor_map.cpp | 52 VERIFY_IS_EQUAL(vec1.dimension(0), 6); in test_1d()
93 VERIFY_IS_EQUAL(mat3.dimension(0), 2); in test_2d()
94 VERIFY_IS_EQUAL(mat3.dimension(1), 3); in test_2d()
98 VERIFY_IS_EQUAL(mat4.dimension(0), 2); in test_2d()
99 VERIFY_IS_EQUAL(mat4.dimension(1), 3); in test_2d()
137 VERIFY_IS_EQUAL(mat3.dimension(0), 2); in test_3d()
138 VERIFY_IS_EQUAL(mat3.dimension(1), 3); in test_3d()
139 VERIFY_IS_EQUAL(mat3.dimension(2), 7); in test_3d()
143 VERIFY_IS_EQUAL(mat4.dimension(0), 2); in test_3d()
144 VERIFY_IS_EQUAL(mat4.dimension(1), 3); in test_3d()
[all …]
|
| D | cxx11_tensor_fft.cpp | 79 VERIFY_IS_EQUAL(forward_output_both_parts.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
80 VERIFY_IS_EQUAL(reverse_output_both_parts.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
82 VERIFY_IS_EQUAL(forward_output_real_part.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
83 VERIFY_IS_EQUAL(reverse_output_real_part.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
85 VERIFY_IS_EQUAL(forward_output_imag_part.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
86 VERIFY_IS_EQUAL(reverse_output_imag_part.dimension(0), input.dimension(0)); in test_fft_complex_input_golden()
136 VERIFY_IS_EQUAL(forward_output_both_parts.dimension(0), input.dimension(0)); in test_fft_real_input_golden()
137 VERIFY_IS_EQUAL(reverse_output_both_parts.dimension(0), input.dimension(0)); in test_fft_real_input_golden()
139 VERIFY_IS_EQUAL(forward_output_real_part.dimension(0), input.dimension(0)); in test_fft_real_input_golden()
140 VERIFY_IS_EQUAL(reverse_output_real_part.dimension(0), input.dimension(0)); in test_fft_real_input_golden()
[all …]
|
| /external/angle/src/libANGLE/renderer/d3d/d3d11/ |
| D | gen_blit11helper.py | 153 def format_shader_include(dimension, blitshader): argument
155 filename=blitshader[2].replace("*", dimension.lower()))
160 for dimension in supported_dimensions:
161 dimension_cases.append(format_get_blitshader_case_dimension(operation, dimension))
168 def format_get_blitshader_case_dimension(operation, dimension): argument
170 if dimension == "2D" and operation.find("TOI") != -1:
173 blitshader = dimension.upper() + "_" + operation
176 dimension=dimension.upper(), blitshader=blitshader)
179 def format_map_blitshader_case(dimension, blitshader): argument
180 blitshader_name = "BLITSHADER_" + dimension.upper() + "_" + blitshader[0]
[all …]
|
| /external/javaparser/javaparser-core/src/main/java/com/github/javaparser/ast/ |
| D | ArrayCreationLevel.java | 48 private Expression dimension; field in ArrayCreationLevel
56 public ArrayCreationLevel(int dimension) { in ArrayCreationLevel() argument
57 this(null, new IntegerLiteralExpr("" + dimension), new NodeList<>()); in ArrayCreationLevel()
60 public ArrayCreationLevel(Expression dimension) { in ArrayCreationLevel() argument
61 this(null, dimension, new NodeList<>()); in ArrayCreationLevel()
65 public ArrayCreationLevel(Expression dimension, NodeList<AnnotationExpr> annotations) { in ArrayCreationLevel() argument
66 this(null, dimension, annotations); in ArrayCreationLevel()
73 …public ArrayCreationLevel(TokenRange tokenRange, Expression dimension, NodeList<AnnotationExpr> an… in ArrayCreationLevel() argument
75 setDimension(dimension); in ArrayCreationLevel()
99 public ArrayCreationLevel setDimension(final Expression dimension) { in setDimension() argument
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 312 inline int32 GetTensorDimIndex(TensorFormat format, char dimension) { in GetTensorDimIndex() argument
315 switch (dimension) { in GetTensorDimIndex()
324 LOG(FATAL) << "Invalid dimension: " << dimension; in GetTensorDimIndex()
328 switch (dimension) { in GetTensorDimIndex()
337 LOG(FATAL) << "Invalid dimension: " << dimension; in GetTensorDimIndex()
341 switch (dimension) { in GetTensorDimIndex()
350 LOG(FATAL) << "Invalid dimension: " << dimension; in GetTensorDimIndex()
354 switch (dimension) { in GetTensorDimIndex()
363 LOG(FATAL) << "Invalid dimension: " << dimension; in GetTensorDimIndex()
381 char dimension) { in GetFilterDimIndex() argument
[all …]
|
| /external/apache-commons-math/src/main/java/org/apache/commons/math/ode/ |
| D | FirstOrderConverter.java | 64 private final int dimension; field in FirstOrderConverter
81 dimension = equations.getDimension(); in FirstOrderConverter()
82 z = new double[dimension]; in FirstOrderConverter()
83 zDot = new double[dimension]; in FirstOrderConverter()
84 zDDot = new double[dimension]; in FirstOrderConverter()
93 return 2 * dimension; in getDimension()
107 System.arraycopy(y, 0, z, 0, dimension); in computeDerivatives()
108 System.arraycopy(y, dimension, zDot, 0, dimension); in computeDerivatives()
114 System.arraycopy(zDot, 0, yDot, 0, dimension); in computeDerivatives()
115 System.arraycopy(zDDot, 0, yDot, dimension, dimension); in computeDerivatives()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | shapes.md | 14 The dimension numbers are arbitrary labels for convenience. The order of
15 these dimension numbers does not imply a particular minor/major ordering in
18 * By convention, dimensions are listed in increasing order of dimension
20 dimension 0 has size `A`, dimension 1 has size `B` and dimension 2 has size
24 dimension -1 is the last dimension (equivalent to `N-1` for an `N`
26 above, dimension -1 has size `C`, dimension -2 has size `B` and so on.
31 * dimension 0: `y`
32 * dimension 1: `x`
36 * dimension 0: `z`
37 * dimension 1: `y`
[all …]
|
| /external/apache-commons-math/src/main/java/org/apache/commons/math/util/ |
| D | MultidimensionalCounter.java | 49 private final int dimension; field in MultidimensionalCounter
74 private final int[] counter = new int[dimension];
92 for (int i = 0; i < dimension; i++) { in hasNext()
131 … return /* Arrays.*/ copyOf(counter, dimension); // Java 1.5 does not support Arrays.copyOf() in getCounts()
165 dimension = size.length; in MultidimensionalCounter()
166 … this.size = /* Arrays.*/ copyOf(size, dimension); // Java 1.5 does not support Arrays.copyOf() in MultidimensionalCounter()
168 uniCounterOffset = new int[dimension]; in MultidimensionalCounter()
170 last = dimension - 1; in MultidimensionalCounter()
174 for (int j = i + 1; j < dimension; j++) { in MultidimensionalCounter()
204 return dimension; in getDimension()
[all …]
|
| /external/webrtc/webrtc/modules/audio_processing/vad/ |
| D | gmm.cc | 24 int dimension, in RemoveMean() argument
26 for (int n = 0; n < dimension; ++n) in RemoveMean()
32 int dimension) { in ComputeExponent() argument
34 for (int i = 0; i < dimension; ++i) { in ComputeExponent()
36 for (int j = 0; j < dimension; j++) in ComputeExponent()
45 if (gmm_parameters.dimension > kMaxDimension) { in EvaluateGmm()
54 RemoveMean(x, mean_vec, gmm_parameters.dimension, v); in EvaluateGmm()
55 double q = ComputeExponent(v, covar_inv, gmm_parameters.dimension) + in EvaluateGmm()
58 mean_vec += gmm_parameters.dimension; in EvaluateGmm()
59 covar_inv += gmm_parameters.dimension * gmm_parameters.dimension; in EvaluateGmm()
|
| /external/apache-commons-math/src/main/java/org/apache/commons/math/random/ |
| D | UnitSphereRandomVectorGenerator.java | 39 private final int dimension; field in UnitSphereRandomVectorGenerator
45 public UnitSphereRandomVectorGenerator(final int dimension, in UnitSphereRandomVectorGenerator() argument
47 this.dimension = dimension; in UnitSphereRandomVectorGenerator()
56 public UnitSphereRandomVectorGenerator(final int dimension) { in UnitSphereRandomVectorGenerator() argument
57 this(dimension, new MersenneTwister()); in UnitSphereRandomVectorGenerator()
63 final double[] v = new double[dimension]; in nextVector()
68 for (int i = 0; i < dimension; i++) { in nextVector()
76 for (int i = 0; i < dimension; i++) { in nextVector()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_dimension_inference.cc | 106 HloInstruction* operand, ShapeIndex index, int64 dimension,
130 hlo, [&](HloInstruction* operand, ShapeIndex index, int64 dimension, in DefaultAction()
135 "@", index.ToString(), "@", dimension, " to hlo ", hlo->ToString(), in DefaultAction()
143 hlo, [&](HloInstruction* operand, ShapeIndex index, int64 dimension, in HandleGetTupleElement() argument
149 parent_->SetDynamicSize(hlo, new_index, dimension, dynamic_size, in HandleGetTupleElement()
158 hlo, [&](HloInstruction*, ShapeIndex index, int64 dimension, in HandleTuple() argument
162 parent_->SetDynamicSize(hlo, index, dimension, dynamic_size, in HandleTuple()
170 hlo, [&](HloInstruction* operand, ShapeIndex index, int64 dimension, in HandleBroadcast() argument
173 int64 broadcast_dim = hlo->dimensions(dimension); in HandleBroadcast()
198 hlo, [&](HloInstruction* operand, ShapeIndex index, int64 dimension, in HandleCustomCall()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 357 EXPECT_EQ(2, tensor.dimension(0)); in SetUp()
358 EXPECT_EQ(3, tensor.dimension(1)); in SetUp()
359 EXPECT_EQ(4, tensor.dimension(2)); in SetUp()
360 EXPECT_EQ(5, tensor.dimension(3)); in SetUp()
388 EXPECT_EQ(*iter, shaped.dimension(i)); in TestReshapeImpl()
401 coord[i] = shaped.dimension(i) - 1; in TestReshapeImpl()
496 EXPECT_EQ(120, flat.dimension(0)); in TEST_F()
506 EXPECT_EQ(24, flat_inner_dims.dimension(0)); in TEST_F()
507 EXPECT_EQ(5, flat_inner_dims.dimension(1)); in TEST_F()
513 EXPECT_EQ(6, flat_inner_dims.dimension(0)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions_test.cc | 51 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
52 EXPECT_EQ(input_backward.dimension(1), input_rows); in TEST()
53 EXPECT_EQ(input_backward.dimension(2), input_cols); in TEST()
103 EXPECT_EQ(input_backward.dimension(0), input_cols); in TEST()
104 EXPECT_EQ(input_backward.dimension(1), input_rows); in TEST()
105 EXPECT_EQ(input_backward.dimension(2), input_depth); in TEST()
151 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
152 EXPECT_EQ(input_backward.dimension(1), input_rows); in TEST()
153 EXPECT_EQ(input_backward.dimension(2), input_cols); in TEST()
202 EXPECT_EQ(input_backward.dimension(0), input_cols); in TEST()
[all …]
|
| D | eigen_pooling_test.cc | 48 EXPECT_EQ(result.dimension(0), depth); in TEST()
49 EXPECT_EQ(result.dimension(1), output_rows); in TEST()
50 EXPECT_EQ(result.dimension(2), output_cols); in TEST()
51 EXPECT_EQ(result.dimension(3), num_batches); in TEST()
97 EXPECT_EQ(result.dimension(3), depth); in TEST()
98 EXPECT_EQ(result.dimension(2), output_rows); in TEST()
99 EXPECT_EQ(result.dimension(1), output_cols); in TEST()
100 EXPECT_EQ(result.dimension(0), num_batches); in TEST()
150 EXPECT_EQ(result.dimension(0), channels); in TEST()
151 EXPECT_EQ(result.dimension(1), output_planes); in TEST()
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 57 EXPECT_EQ(input_backward.dimension(3), input_cols); in TEST()
58 EXPECT_EQ(input_backward.dimension(2), input_rows); in TEST()
59 EXPECT_EQ(input_backward.dimension(1), input_planes); in TEST()
60 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
120 EXPECT_EQ(input_backward.dimension(0), input_cols); in TEST()
121 EXPECT_EQ(input_backward.dimension(1), input_rows); in TEST()
122 EXPECT_EQ(input_backward.dimension(2), input_planes); in TEST()
123 EXPECT_EQ(input_backward.dimension(3), input_depth); in TEST()
183 EXPECT_EQ(input_backward.dimension(3), input_cols); in TEST()
184 EXPECT_EQ(input_backward.dimension(2), input_rows); in TEST()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 185 const int batch = input.dimension(0); in operator ()()
186 const int input_rows = input.dimension(1); in operator ()()
187 const int input_cols = input.dimension(2); in operator ()()
188 const int depth = input.dimension(3); in operator ()()
190 const int filter_rows = filter.dimension(0); in operator ()()
191 const int filter_cols = filter.dimension(1); in operator ()()
193 const int output_rows = output.dimension(1); in operator ()()
194 const int output_cols = output.dimension(2); in operator ()()
216 const int batch = input.dimension(0); in operator ()()
217 const int input_rows = input.dimension(1); in operator ()()
[all …]
|