Home
last modified time | relevance | path

Searched defs:extractTensors (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_ivalue.h45 std::vector<at::Tensor> extractTensors() override { in extractTensors() function
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp72 std::vector<at::Tensor> extractTensors(const c10::IValue& result) { in extractTensors() function