Home
last modified time | relevance | path

Searched refs:extractTensors (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/csrc/jit/python/
Dpython_ivalue.h45 std::vector<at::Tensor> extractTensors() override { in extractTensors() function
/external/pytorch/torch/csrc/distributed/c10d/
Dcomm.cpp111 result.toPyObjectHolder()->extractTensors(); in parseCppCommHookResult()
DWork.cpp162 return _fut->value().toPyObjectHolder()->extractTensors(); in result()
Dreducer.cpp72 std::vector<at::Tensor> extractTensors(const c10::IValue& result) { in extractTensors() function
74 return result.toPyObjectHolder()->extractTensors(); in extractTensors()
546 auto results = extractTensors(workHandle->getFuture()->value()); in set_divide_factor()
/external/pytorch/aten/src/ATen/core/
Divalue.cpp1089 value.toPyObjectHolder()->extractTensors(); in extractStorages()
Divalue_inl.h1619 virtual std::vector<at::Tensor> extractTensors() = 0;