| cxx11_tensor_image_patch.cpp | 188 int input_batches = 1; local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
203 VERIFY_IS_EQUAL(result.dimension(4), input_batches); // number of batches
229 for (int b = 0; b < input_batches; ++b) { // batch
264 int input_batches = 2; local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
276 VERIFY_IS_EQUAL(result.dimension(4), input_batches); // number of batches
302 for (int b = 0; b < input_batches; ++b) { // batch
337 int input_batches = 1; local
341 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
[all...] |
