Searched defs:chunk_sizes (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | comm.cpp | 243 std::vector<int64_t> chunk_sizes; in _broadcast_out_impl() local
316 const std::optional<std::vector<int64_t>>& chunk_sizes, in _broadcast_out_impl()
378 std::vector<int64_t> chunk_sizes; in _broadcast_out_impl() local
|
| H A D | python_comm.cpp | 51 std::optional<py::object> py_streams) { in initCommMethods()
|
| /aosp_15_r20/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/crypto/cipher_extra/ |
| D | cipher_test.cc | 350 const std::vector<size_t> chunk_sizes = {0, 1, 2, 5, 7, 8, 9, 15, 16, in TestCipher() local
455 const std::vector<size_t> chunk_sizes = { in TEST() local
|
| /aosp_15_r20/external/rust/crates/quiche/deps/boringssl/src/crypto/cipher_extra/ |
| D | cipher_test.cc | 350 const std::vector<size_t> chunk_sizes = {0, 1, 2, 5, 7, 8, 9, 15, 16, in TestCipher() local
455 const std::vector<size_t> chunk_sizes = { in TEST() local
|
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | comm.py | 168 def scatter(tensor, devices=None, chunk_sizes=None, dim=0, streams=None, *, out=None): argument
|
| H A D | _functions.py | 94 def forward(ctx, target_gpus, chunk_sizes, dim, input): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/ |
| H A D | comm.cpp | 18 std::optional<std::vector<int64_t>> chunk_sizes, in Scatter()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cuda_multigpu.py | 1502 def _test_scatter(self, input, chunk_sizes=None, dim=0): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | dataloader.cpp | 94 size_t chunk_sizes[chunk_count_] = {10, 5, 20}; member
|