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...] |