Searched defs:Functionalize (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | Interpreter.h | 56 Functionalize, enumerator
134 static Interpreter Functionalize(int64_t level, bool functionalizeAddBackViews) { in Functionalize() function
|
| /external/pytorch/torch/csrc/inductor/ |
| D | resize_storage_bytes.cpp | 60 TORCH_LIBRARY_IMPL(inductor, Functionalize, m) { in TORCH_LIBRARY_IMPL() argument
|
| /external/pytorch/torch/onnx/_internal/fx/passes/ |
| D | functionalization.py | 18 class Functionalize(_pass.Transform): class
|
| /external/pytorch/aten/src/ATen/templates/ |
| D | RegisterFunctionalization.cpp | 104 TORCH_LIBRARY_IMPL(aten, Functionalize, m) { in TORCH_LIBRARY_IMPL() argument
|
| /external/pytorch/aten/src/ATen/ |
| D | FunctionalizeFallbackKernel.cpp | 349 TORCH_LIBRARY_IMPL(_, Functionalize, m) { in TORCH_LIBRARY_IMPL() argument
353 TORCH_LIBRARY_IMPL(aten, Functionalize, m) { in TORCH_LIBRARY_IMPL() argument
|
| /external/pytorch/c10/core/ |
| D | DispatchKey.h | 246 Functionalize, enumerator
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 1598 Status FunctionalizeCond::Functionalize(Graph* graph, in Functionalize() function in tensorflow::functionalize_cond::FunctionalizeCond
|
| /external/pytorch/torchgen/ |
| D | model.py | 122 Functionalize = auto() variable in DispatchKey
|