Searched defs:get_batch (Results 1 – 9 of 9) sorted by relevance
67 BatchType get_batch() { in get_batch() function354 BatchType get_batch(size_t batch_size) override { in get_batch() function370 BatchType get_batch() { in get_batch() function
36 BatchType get_batch(BatchRequestType request) override { in get_batch() function
41 OutputBatchType get_batch(BatchRequestType indices) override { in get_batch() function
146 std::vector<int> get_batch(size_t batch_size) override { in get_batch() function802 std::vector<int> get_batch(TestIndex index) override { in get_batch() function1149 int get_batch(torch::ArrayRef<size_t> indices) override { in TEST() function1373 size_t get_batch(torch::ArrayRef<size_t> indices) override { in get_batch() function1470 torch::optional<int> get_batch(size_t) override { in TEST() function1507 torch::optional<int> get_batch(size_t) override { in TEST() function1547 torch::optional<int> get_batch(size_t) override { in TEST() function1590 torch::optional<std::vector<Example<>>> get_batch( in TEST() function
58 def get_batch(): function
66 def get_batch(): function
62 def get_batch(): function
36 pub(super) fn get_batch(input: &[u8], n: usize) -> u32 { in get_batch() function42 pub(super) fn get_batch(input: &[u8], n: usize) -> u32 { in get_batch() function
4089 def get_batch(): function