| /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/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 | 124 int64_t end_dim) { in flatten_copy_out()
1715 const auto end_dim = p_node->Input(2).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/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 | 9228 int end_dim = negative_end_dim ? pos_end_dim - rank : pos_end_dim; in TEST_F() local
|