Home
last modified time | relevance | path

Searched defs:unflatten (Results 1 – 21 of 21) sorted by relevance

/external/pytorch/torch/_library/
Dautograd.py225 unflatten = _pytree.tree_unflatten variable
/external/executorch/extension/pytree/
Dpybindings.cpp41 py::function unflatten; member
57 py::function unflatten) { in register_custom_type()
Dpytree.h651 ContainerHandle<T, Aux> unflatten(const TreeSpec<Aux>& spec, T* leaves) { in unflatten() function
659 ContainerHandle<T, Aux> unflatten(const StrTreeSpec& spec, T* leaves) { in unflatten() function
/external/executorch/extension/pytree/aten_util/
Divalue_util.cpp141 IValue unflatten( in unflatten() function
/external/pytorch/torch/csrc/jit/python/
Dpython_arg_flatten.cpp182 PyObject* unflatten(ArrayRef<Variable> vars, const IODescriptor& desc) { in unflatten() function
/external/pytorch/torch/_functorch/_aot_autograd/
Dutils.py157 def unflatten(self, x: List[Any]) -> Any: member in PytreeThunk
/external/skia/src/core/
DSkImageFilter.cpp177 bool SkImageFilter_Base::Common::unflatten(SkReadBuffer& buffer, int expectedCount) { in unflatten() function in SkImageFilter_Base::Common
/external/executorch/runtime/executor/test/
Dexecutor_test.cpp239 auto unflatten(EValue* items) { in unflatten() function
/external/pytorch/torch/export/
Dunflatten.py529 def unflatten( function
/external/pytorch/torch/distributed/_symmetric_memory/
D__init__.py319 def unflatten(t: torch.Tensor) -> torch.Tensor: function
/external/pytorch/torch/onnx/
Dsymbolic_opset13.py457 def unflatten(g: jit_utils.GraphContext, input, dim, unflattened_size): function
/external/pytorch/torch/
D_tensor.py1356 def unflatten(self, dim, sizes): member in Tensor
/external/skia/src/shaders/gradients/
DSkGradientBaseShader.cpp131 bool SkGradientBaseShader::DescriptorScope::unflatten(SkReadBuffer& buffer, in unflatten() function in SkGradientBaseShader::DescriptorScope
/external/executorch/backends/apple/coreml/runtime/inmemoryfs/
Dinmemory_filesystem.cpp358 FlattenedInMemoryNode::unflatten(const std::vector<FlattenedInMemoryNode>& flattened_nodes, in unflatten() function in __anon6b896f7c0111::FlattenedInMemoryNode
/external/pytorch/functorch/csrc/dim/
Ddim.cpp1033 Unflatten unflatten; member
1079 auto unflatten = mpy::object::checked_steal(PyCFunction_New(&py_unflatten_def, cap.release())); in py_tree_flatten() local
1091 auto unflatten = tree_flatten(A, agg, elements); in tree_map() local
/external/pytorch/torch/utils/
D_pytree.py801 def unflatten(self, leaves: Iterable[Any]) -> PyTree: member in TreeSpec
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc1742 HloInstruction* unflatten = reshape->AddInstruction( in RewriteDynamicReshape() local
/external/pytorch/torch/_refs/
D__init__.py3937 def unflatten(a: TensorLikeType, dim: int, sizes: ShapeType) -> TensorLikeType: function
/external/pytorch/test/cpp/api/
Dmodules.cpp607 Unflatten unflatten(UnflattenOptions(0, {2, 2})); in TEST_F() local
/external/icu/icu4c/source/tools/toolutil/
Djson-json.hpp14640 unflatten(const BasicJsonType& value) in unflatten() function in json_pointer
24013 basic_json unflatten() const in unflatten() function in basic_json
/external/ublksrv/include/nlohmann/
Djson.hpp14242 unflatten(const BasicJsonType& value) in unflatten() function in nlohmann::json_pointer
23436 basic_json unflatten() const in unflatten() function in nlohmann::basic_json