Home
last modified time | relevance | path

Searched refs:tensorTable (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/rpc_messages/
H A Dcleanup_autograd_context_req.cpp18 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
20 jit::pickle(at::IValue(context_id_), &tensorTable); in toMessageImpl()
23 std::move(tensorTable), in toMessageImpl()
H A Drref_backward_req.cpp29 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
31 jit::pickle(c10::ivalue::Tuple::create(std::move(ivalues)), &tensorTable); in toMessageImpl()
35 std::move(tensorTable), in toMessageImpl()
H A Dpropagate_gradients_req.cpp39 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
41 jit::pickle(c10::ivalue::Tuple::create(std::move(ivalues)), &tensorTable); in toMessageImpl()
45 std::move(tensorTable), in toMessageImpl()
H A Drpc_with_autograd.cpp76 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
78 jit::pickle(c10::ivalue::Tuple::create(std::move(ivalues)), &tensorTable); in toMessageImpl()
81 TORCH_INTERNAL_ASSERT(tensorTable.empty()); in toMessageImpl()
H A Drpc_with_profiling_req.cpp77 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
79 jit::pickle(c10::ivalue::Tuple::create(std::move(ivalues)), &tensorTable); in toMessageImpl()
H A Drpc_with_profiling_resp.cpp84 std::vector<torch::Tensor> tensorTable; in toMessageImpl() local
86 jit::pickle(c10::ivalue::Tuple::create(std::move(ivalues)), &tensorTable); in toMessageImpl()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dutils.cpp512 std::vector<torch::Tensor> tensorTable; in readWrappedPayload() local
517 tensorTable); in readWrappedPayload()