Searched full:dtype_layout (Results 1 – 13 of 13) sorted by relevance
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | symbolic_shape_registry_util.cpp | 28 …{"aten::to.dtype_layout(Tensor self, *, ScalarType? dtype=None, Layout? layout=None, Device? devic… in get_tensorexpr_elementwise_set()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDecompositions.cpp | 388 OP_DECOMPOSE2(to, dtype_layout); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 363 aten::to.dtype_layout: 79
|
| D | coverage.yaml | 595 - aten::to.dtype_layout
989 aten::to.dtype_layout: 27
|
| /external/pytorch/aten/src/ATen/ |
| D | FunctionalizeFallbackKernel.cpp | 234 // note I only need this because the to.dtype/to.dtype_layout overload calls this, so we skip the o…
|
| D | LegacyBatchingRegistrations.cpp | 1187 m.impl("to.dtype_layout", to_dtype_layout_batching_rule); in TORCH_LIBRARY_IMPL()
|
| /external/executorch/backends/cadence/aot/tests/ |
| D | test_remove_ops_passes.py | 67 count_node(graph_after_passes, exir_ops.edge.aten.to.dtype_layout),
|
| /external/executorch/backends/cadence/aot/ |
| D | remove_ops.py | 159 exir_ops.edge.aten.to.dtype_layout,
|
| /external/pytorch/torch/_subclasses/ |
| D | functional_tensor.py | 376 if func == torch.ops.aten.to.dtype_layout:
|
| /external/pytorch/torch/export/ |
| D | exported_program.py | 284 (torch.ops.aten.to.dtype_layout, torch.ops.aten.to.dtype),
|
| /external/pytorch/aten/src/ATen/core/ |
| D | NamedRegistrations.cpp | 484 m.impl("to.dtype_layout", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | lowerings.cpp | 1576 …"aten::to.dtype_layout(Tensor(a) self, *, int? dtype=None, int? layout=None, Device? device=None, … in nnc_lowerings_lazy_registration()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | native_functions.yaml | 7669 - func: to.dtype_layout(Tensor(a) self, *, ScalarType? dtype=None, Layout? layout=None, Device? dev…
|