Home
last modified time | relevance | path

Searched defs:step_buffers (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dserialize.cpp562 std::vector<int64_t> step_buffers; in TEST() local
654 std::vector<int64_t> step_buffers; in TEST() local
713 std::vector<int64_t> step_buffers; in TEST() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/
H A Dadagrad.cpp140 std::vector<int64_t> step_buffers; in load() local
H A Dadam.cpp158 std::vector<int64_t> step_buffers; in load() local
H A Dadamw.cpp159 std::vector<int64_t> step_buffers; in load() local