Home
last modified time | relevance | path

Searched refs:ArrayRefCType (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/tools/test/
Dtest_executorch_types.py18 from torchgen.executorch.api.types import ArrayRefCType, scalarT, tensorListT, tensorT
47 ("int[] size", NamedCType("size", ArrayRefCType(BaseCType(longT)))),
52 NamedCType("spacing", ArrayRefCType(ConstRefCType(BaseCType(scalarT)))),
56 NamedCType("weight", ArrayRefCType(OptionalCType(BaseCType(tensorT)))),
61 "output_size", OptionalCType(ArrayRefCType(BaseCType(longT)))
66 NamedCType("dims", OptionalCType(ArrayRefCType(BaseCType(longT)))),
70 NamedCType("output_mask", ArrayRefCType(BaseCType(boolT))),
Dtest_executorch_unboxing.py76 self.assertIsInstance(ctype, et_types.ArrayRefCType)
79 ctype, et_types.ArrayRefCType(elem=BaseCType(aten_types.longT))
/external/pytorch/torchgen/executorch/api/types/
Dtypes.py72 class ArrayRefCType(CType): class
83 return ArrayRefCType(self.elem.remove_const_ref())
/external/pytorch/torchgen/executorch/api/
Det_cpp.py20 ArrayRefCType,
90 binds, ArrayRefCType(BaseCType(BaseTypeToCppMapping[BaseTy.bool]))
148 return NamedCType(binds, ArrayRefCType(OptionalCType(BaseCType(tensorT))))
150 return NamedCType(binds, ArrayRefCType(elem.type))
/external/pytorch/torchgen/api/types/
Dtypes.py162 class ArrayRefCType(CType): class
173 return ArrayRefCType(self.elem.remove_const_ref())
/external/pytorch/torchgen/api/
Dstructured.py6 ArrayRefCType,
84 return NamedCType(binds, ArrayRefCType(elem.type))
Dcpp.py9 ArrayRefCType,
209 return NamedCType(binds, ArrayRefCType(BaseCType(scalarT)))
217 return NamedCType(binds, ArrayRefCType(elem.type))
Dtranslate.py6 ArrayRefCType,
259 if goal.type == ArrayRefCType(BaseCType(longT)):
/external/pytorch/tools/autograd/
Dgen_autograd_functions.py20 ArrayRefCType,
697 elif type == OptionalCType(ArrayRefCType(BaseCType(doubleT))):
732 elif type == ArrayRefCType(
Dgen_variable_type.py45 ArrayRefCType,
1509 elif type == ArrayRefCType(