Searched defs:idx_shape (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ |
| D | test_mps.py | 7401 def helper(shape, dim, idx_shape, idx_dtype=torch.int64): argument
7456 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, do_add=True): argument
7563 def helper(shape, dim, idx_shape, src_shape, idx_dtype=torch.int64, reduce_str="sum"): argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 1931 auto idx_shape = std::vector<int64_t>(t.dim(), 1); in index_select_sparse_cpu() local
|