Lines Matching defs:batch
30 const ptrdiff_t batch = 10;
34 Tensor<float, 4> input(depth, rows, cols, batch);
37 Tensor<float, 4> result(depth, rows, cols, batch);
40 for (int b = 0; b < batch; ++b) {
54 const ptrdiff_t batch = 10;
58 Tensor<float, 4> input(depth, rows, cols, batch);
61 Tensor<float, 4> result(depth, rows, cols, batch);
64 for (int b = 0; b < batch; ++b) {
78 const ptrdiff_t batch = 10;
82 Tensor<float, 4> input(depth, rows, cols, batch);
85 Tensor<float, 4> result(depth, rows, cols, batch);
88 for (int b = 0; b < batch; ++b) {
