Home
last modified time | relevance | path

Searched defs:MemoryFormat (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/android/pytorch_android/src/main/java/org/pytorch/
H A DMemoryFormat.java3 public enum MemoryFormat { enum
11 MemoryFormat(int jniCode) { in MemoryFormat() method in MemoryFormat
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DRandom.cpp68 const std::optional<c10::MemoryFormat> /* not implemented */) { in rand_like()
129 const std::optional<c10::MemoryFormat> /* not implemented */) { in randn_like()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/
H A Dexec_aten.h69 using MemoryFormat = c10::MemoryFormat; variable
113 using MemoryFormat = torch::executor::MemoryFormat; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorFactories.cpp114 std::optional<c10::MemoryFormat>) { in empty_affine_quantized_other_backends_stub()
127 std::optional<c10::MemoryFormat>) { in empty_per_channel_affine_quantized_other_backends_stub()
/aosp_15_r20/external/executorch/runtime/core/portable_type/
H A Dtensor_options.h22 enum class MemoryFormat : int8_t { enum
/aosp_15_r20/external/pytorch/c10/core/
H A DMemoryFormat.h29 enum class MemoryFormat : int8_t { enum
/aosp_15_r20/external/executorch/kernels/aten/cpu/
H A Dop__to_dim_order_copy.cpp19 using MemoryFormat = exec_aten::MemoryFormat; typedef
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop__to_dim_order_copy.cpp21 using MemoryFormat = exec_aten::MemoryFormat; typedef
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dschema.py43 class MemoryFormat(IntEnum): class
/aosp_15_r20/external/executorch/exir/serde/
H A Dschema.py51 class MemoryFormat(IntEnum): class
/aosp_15_r20/external/executorch/runtime/core/test/
H A Devalue_test.cpp173 TEST_F(EValueTest, MemoryFormat) { in TEST_F() argument
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py1890 MemoryFormat = auto() variable in BaseTy