Searched full:unsqueeze_ (Results 1 – 25 of 63) sorted by relevance
123
| /external/pytorch/aten/src/ATen/native/ |
| D | AffineGridGenerator.cpp | 40 base_grid.select(-1, 1).copy_(linspace_from_neg_one(theta, H, align_corners).unsqueeze_(-1)); in make_base_grid_4D()
57 base_grid.select(-1, 1).copy_(linspace_from_neg_one(theta, H, align_corners).unsqueeze_(-1)); in make_base_grid_5D()
58 ….select(-1, 2).copy_(linspace_from_neg_one(theta, D, align_corners).unsqueeze_(-1).unsqueeze_(-1)); in make_base_grid_5D()
|
| D | SortingUtils.h | 34 values.unsqueeze_(dim); in _reduction_with_indices_allocate_or_resize_output()
48 indices.unsqueeze_(dim); in _reduction_with_indices_allocate_or_resize_output()
|
| D | TensorConversions.cpp | 657 compressed_indices.unsqueeze_(0); in sparse_compressed_to_dense()
658 plain_indices.unsqueeze_(0); in sparse_compressed_to_dense()
659 values.unsqueeze_(0); in sparse_compressed_to_dense()
660 dense.unsqueeze_(0); in sparse_compressed_to_dense()
1279 compressed_indices.unsqueeze_(0); in sparse_compressed_to_flipped()
1280 plain_indices.unsqueeze_(0); in sparse_compressed_to_flipped()
1281 values.unsqueeze_(0); in sparse_compressed_to_flipped()
1890 .mul(at::tensor(batch_blocksize, indices.options()).unsqueeze_(1)) in sparse_compressed_to_sparse()
1897 .unsqueeze_(-1).add(block_coo_indices.unsqueeze_(1)) in sparse_compressed_to_sparse()
|
| D | MathBitFallThroughLists.h | 32 m.impl("unsqueeze_", torch::CppFunction::makeFallthrough()); \
|
| D | SparseTensorUtils.cpp | 137 a.unsqueeze_(j); in full_coo_indices()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorModeKernel.cpp | 34 values.unsqueeze_(dim); in mode_kernel_impl()
37 indices.unsqueeze_(dim); in mode_kernel_impl()
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseBlasImpl.cpp | 128 return t.unsqueeze(-1).unsqueeze_(-1); in _compressed_row_strided_mm_out()
175 values.unsqueeze_(-1).unsqueeze_(-1); in _compressed_row_strided_mm_out()
|
| D | ValidateCompressedIndicesCommon.h | 293 at::arange(batch_count, cidx.options()).view(batch_dims).unsqueeze_(-1); in _validate_compressed_sparse_indices_kernel()
|
| /external/pytorch/aten/src/ATen/test/ |
| D | scalar_tensor_test.cpp | 74 // unsqueeze_ in test()
77 auto r = t2.unsqueeze_(0); in test()
|
| /external/executorch/backends/xnnpack/test/ops/ |
| D | sdpa.py | 58 mask.unsqueeze_(0)
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | TensorCompareKernel.cpp | 46 result1.unsqueeze_(dim); in compare_base_kernel_core()
49 result2.unsqueeze_(dim); in compare_base_kernel_core()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | LegacyBatchingRegistrations.cpp | 174 // We "fix it up" by calling unsqueeze_. in squeeze__batching_rule()
188 return self.unsqueeze_(dim); in unsqueeze__batching_rule()
198 batched->value().unsqueeze_(dim_physical); in unsqueeze__batching_rule()
749 m.impl("unsqueeze_", unsqueeze__batching_rule); in TORCH_LIBRARY_IMPL()
|
| D | BatchedTensorImpl.h | 96 // the index of the bdim (think squeeze_, unsqueeze_)
|
| D | BatchRulesFactory.cpp | 130 …result = (start_ + at::arange(0, steps, tensor_options).unsqueeze_(1) * (end_ - start_) / (steps -… in linspace_logspace_batch_rule_helper()
|
| D | BatchRulesScatterOps.cpp | 1081 self_.unsqueeze_(-1); in index_fill_batch_rule_helper()
1106 self_.unsqueeze_(-1); in index_fill_int_scalar_batch_rule_impl()
1158 self_.unsqueeze_(-1); in index_fill_int_tensor_batch_rule_impl()
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | dropout.h | 67 input = input.unsqueeze_(0); in _dropoutNd_helper()
|
| /external/pytorch/torch/ao/ns/fx/ |
| D | mappings.py | 735 "unsqueeze_",
|
| /external/pytorch/torch/optim/ |
| D | adamax.py | 281 [exp_inf.mul_(beta2).unsqueeze(0), grad.abs().add_(eps).unsqueeze_(0)],
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 376 aten::unsqueeze_: 12
|
| D | coverage.yaml | 617 - aten::unsqueeze_
999 aten::unsqueeze_: 6
|
| /external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| D | Blas.cpp | 221 binary = binary.dim() < 3 ? binary.unsqueeze_(0) : binary; in baddbmm_out()
|
| /external/pytorch/docs/source/ |
| D | tensors.rst | 772 Tensor.unsqueeze_
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | _common_operator_config_utils.py | 680 "unsqueeze_",
|
| /external/pytorch/torch/testing/_internal/ |
| D | composite_compliance.py | 92 'unsqueeze_',
|
| /external/pytorch/torch/ao/quantization/ |
| D | utils.py | 114 "unsqueeze_",
|
123