Lines Matching defs:batch
30 const ptrdiff_t batch = 10;
36 Tensor<float, 4> input(depth, rows, cols, batch);
40 offsets.resize(batch);
41 for (int i = 0; i < batch; ++i) {
46 Tensor<float, 4> result(depth, glimpse_rows, glimpse_cols, batch);
49 for (int b = 0; b < batch; ++b) {
66 const ptrdiff_t batch = 10;
72 Tensor<float, 4> input(depth, rows, cols, batch);
76 offsets.resize(batch);
77 for (int i = 0; i < batch; ++i) {
82 Tensor<float, 4> result(depth, glimpse_rows, glimpse_cols, batch);
85 for (int b = 0; b < batch; ++b) {
