Searched refs:max_shapes (Results 1 – 2 of 2) sorted by relevance
177 std::vector<TensorShape> max_shapes; // Of non-queue components. in GetNextInternal() local195 max_shapes.push_back(std::move(out_shape)); in GetNextInternal()212 for (int i = 0; i < max_shapes.size(); ++i) { in GetNextInternal()213 Tensor component(cpu_allocator(), dtypes[i], max_shapes[i]); in GetNextInternal()226 if (batch[b][i].shape() == max_shapes[i]) { in GetNextInternal()
761 # define max_shapes 256 in main() macro762 struct arg arg_list[max_shapes]; in main()777 for (nshapes=0; 3+7*(nshapes+1) <= argc && nshapes < max_shapes; in main()