Home
last modified time | relevance | path

Searched defs:apply_batch (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/transforms/
H A Dbase.h42 std::vector<Output> apply_batch(std::vector<Input> input_batch) override { in apply_batch() function
H A Dlambda.h26 OutputBatchType apply_batch(InputBatchType input_batch) override { in apply_batch() function
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Ddataloader.cpp261 int apply_batch(std::vector<int> input) override { in TEST() function