| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _view_ops.py | 221 def dim_flatten(ndim: int, start_dim=0, end_dim=-1) -> DimMap: argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | graph_gradual_typechecker.py | 489 def flatten_check(tensor_type, start_dim, end_dim): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 295 def flatten(g: jit_utils.GraphContext, input, start_dim, end_dim): argument
|
| H A D | symbolic_opset11.py | 1164 def flatten(g: jit_utils.GraphContext, input, start_dim, end_dim): argument
|
| H A D | symbolic_helper.py | 1460 def _flatten_helper(g: jit_utils.GraphContext, input, start_dim, end_dim, dim): argument
|
| H A D | symbolic_opset9.py | 4959 def flatten(g: jit_utils.GraphContext, input, start_dim, end_dim): argument
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/migrate_gradual_types/ |
| H A D | constraint_generator.py | 35 def generate_flatten_constraints(start_dim, end_dim, input, flattened, n, counter): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | misc.cpp | 370 auto start_dim = at::maybe_wrap_dim(std::get<int64_t>(inputs[1]), A.ndim()); in computeTranspose() local
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | ops.py | 324 def _flatten_sig(input, start_dim=0, end_dim=-1): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 123 int64_t start_dim, in flatten_copy_out()
1714 const auto start_dim = p_node->Input(1).toInt(); in __anon11f46a8b4802() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 3437 Tensor flatten(const Tensor& self, int64_t start_dim, int64_t end_dim) { in flatten()
3467 Tensor flatten(const Tensor& self, int64_t start_dim, int64_t end_dim, Dimname out_dim) { in flatten()
3485 Tensor flatten(const Tensor& self, Dimname start_dim, Dimname end_dim, Dimname out_dim) { in flatten()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIterator.cpp | 870 void TensorIteratorBase::select_all_keeping_dim(int start_dim, IntArrayRef indices) { in select_all_keeping_dim()
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 1316 [&](std::vector<int64_t> shape, int64_t start_dim, int64_t end_dim) { in TEST()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 11563 def reference_flatten(input, start_dim=0, end_dim=-1): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 9226 int start_dim = in TEST_F() local
|