Home
last modified time | relevance | path

Searched refs:FunctionSchema (Results 1 – 25 of 143) sorted by relevance

123456

/external/pytorch/torch/csrc/jit/runtime/
Dsymbolic_shape_registry.cpp71 std::unordered_map<const FunctionSchema*, std::shared_ptr<Graph>>
74 std::unordered_map<const FunctionSchema*, BoundedShapeGraphs>
80 const std::optional<const FunctionSchema*> getInplaceVariant( in getInplaceVariant()
81 const FunctionSchema& base_schema) { in getInplaceVariant()
88 const FunctionSchema* schema = &variant->schema(); in getInplaceVariant()
121 const FunctionSchema* schema, in checkForWhileLoop()
140 const FunctionSchema* schema, in checkInputReturnedAsOutput()
162 const FunctionSchema* schema, in checkInputAndOutputTypes()
221 const FunctionSchema* schema_string, in transformShapeFunction()
246 const FunctionSchema* schema_string, in genShapeComputeFn()
[all …]
Doperator.h32 using ::c10::FunctionSchema;
76 mutable std::variant<FunctionSchema, UnparsedFunctionSchema> schema_;
101 FunctionSchema(varArgSchemaWithName( in Operator()
126 FunctionSchema(varArgSchemaWithName(name, alias_analysis)), in Operator()
163 const FunctionSchema& schema() const { in schema()
166 [](const C10Operator& op) -> const FunctionSchema& { in schema()
169 [](const JitOnlyOperator& op) -> const FunctionSchema& { in schema()
175 FunctionSchema schema = in schema()
184 return std::get<FunctionSchema>(op.schema_); in schema()
208 const FunctionSchema& schemaRef = schema(); in aliasAnalysisKind()
[all …]
Ddecomposition_registry.cpp26 std::unordered_map<const FunctionSchema*, std::shared_ptr<Graph>>
30 std::unordered_map<const FunctionSchema*, std::unique_ptr<Function>>
33 std::unordered_map<const FunctionSchema*, Function*> schema_to_function;
38 const FunctionSchema* schema = &pair.first->schema(); in loadModule()
111 const FunctionSchema& schema) { in GetDecomposition()
124 const FunctionSchema& schema) { in GetDecompositionFunction()
145 const FunctionSchema& schema, in RegisterDecomposition()
165 bool has_jit_decomposition(const c10::FunctionSchema& schema) const override;
180 bool JitDecomp::has_jit_decomposition(const FunctionSchema& schema) const { in has_jit_decomposition()
200 bool has_jit_decomposition(const FunctionSchema& schema) { in has_jit_decomposition()
[all …]
Dsymbolic_shape_registry.h54 const FunctionSchema& schema,
58 const FunctionSchema& schema);
61 const FunctionSchema& schema);
63 TORCH_API std::vector<const FunctionSchema*> RegisteredShapeComputeSchemas();
66 const FunctionSchema* schema,
Ddecomposition_registry.h11 const FunctionSchema& schema);
14 const FunctionSchema& schema,
20 const FunctionSchema& schema);
25 TORCH_API Function* GetDecompositionExecutor(const FunctionSchema& schema);
31 TORCH_API bool has_jit_decomposition(const FunctionSchema& schema);
Dsymbolic_script.cpp1493 std::unordered_map<const FunctionSchema*, GradientPair> cached_gradient_pairs;
1526 static std::string overloadedSchemaString(const FunctionSchema& schema) { in overloadedSchemaString()
1595 const FunctionSchema& loaded_schema = method->getSchema(); in loadModule()
1596 FunctionSchema actual_schema( in loadModule()
1618 const FunctionSchema& schema) { in gradientInfoForSchema()
1641 bool hasGradientInfoForSchema(const FunctionSchema& schema) { in hasGradientInfoForSchema()
/external/pytorch/aten/src/ATen/core/op_registration/
Dop_registration.cpp40 FunctionSchema inferred_schema = inferSchemaFromKernels_(name, options); in checkSchemaAndRegisterOp_()
42 options.schemaOrName_ = FunctionSchema( in checkSchemaAndRegisterOp_()
57 std::get<FunctionSchema>(options.schemaOrName_.value()), in checkSchemaAndRegisterOp_()
65 c10::FunctionSchema RegisterOperators::inferSchemaFromKernels_( in inferSchemaFromKernels_()
74 std::optional<FunctionSchema> inferred_schema = std::nullopt; in inferSchemaFromKernels_()
103 toString(std::get<FunctionSchema>(options.schemaOrName_.value()))); in checkNoDuplicateKernels_()
109 toString(std::get<FunctionSchema>(options.schemaOrName_.value()))); in checkNoDuplicateKernels_()
116 FunctionSchema schema = in registerOp_()
117 std::get<FunctionSchema>(std::move(options.schemaOrName_.value())); in registerOp_()
Dinfer_schema.h110 TORCH_API FunctionSchema make_function_schema(std::string&& name, std::string&& overload_name, c10:…
111 TORCH_API FunctionSchema make_function_schema(c10::ArrayRef<ArgumentDef> arguments, c10::ArrayRef<A…
116 FunctionSchema createFunctionSchemaFromTraitsFlattenedReturns() {
132 FunctionSchema createFunctionSchemaFromTraitsSingleReturn(std::string&& name, std::string&& overloa…
149 FunctionSchema inferFunctionSchemaFlattenedReturns() {
154 FunctionSchema inferFunctionSchemaSingleReturn(std::string&& name, std::string&& overload_name) {
158 …d::optional<std::string> findSchemaDifferences(const FunctionSchema& inferred, const FunctionSchem…
Dinfer_schema.cpp26 FunctionSchema make_function_schema( in make_function_schema()
31 return FunctionSchema( in make_function_schema()
38 FunctionSchema make_function_schema( in make_function_schema()
47 const FunctionSchema& lhs, in findSchemaDifferences()
48 const FunctionSchema& rhs) { in findSchemaDifferences()
/external/pytorch/aten/src/ATen/core/
Dfunction_schema.h22 struct FunctionSchema;
226 bool operator==(const FunctionSchema& lhs, const FunctionSchema& rhs);
228 struct TORCH_API FunctionSchema { struct
229 FunctionSchema(
244 FunctionSchema(
251 : FunctionSchema(
285 const FunctionSchema& old,
307 const FunctionSchema& old,
441 FunctionSchema cloneWithName(std::string name, std::string overload_name) const { in cloneWithName() argument
442 return FunctionSchema( in cloneWithName()
[all …]
Dfunction_schema.cpp9 void FunctionSchema::dump() const { in dump()
13 const std::vector<Argument>& FunctionSchema::getCorrectList(SchemaArgType type) const { in getCorrectList()
21 FunctionSchema FunctionSchema::cloneWithRealTypes(bool with_symint) const { in cloneWithRealTypes()
47 return FunctionSchema( in cloneWithRealTypes()
56 bool FunctionSchema::canAliasTypeSetsAlias(const std::optional<AliasTypeSet> &lhs, const std::optio… in canAliasTypeSetsAlias()
70 std::optional<AliasTypeSet> FunctionSchema::getAliasTypeSetContainedTypes(const std::optional<Alias… in getAliasTypeSetContainedTypes()
98 std::optional<AliasTypeSet> FunctionSchema::mapTypeToAliasTypeSet(const TypePtr& type) const { in mapTypeToAliasTypeSet()
147 bool FunctionSchema::may_alias(const SchemaArgument& lhs, const SchemaArgument& rhs) const { in may_alias()
177 bool FunctionSchema::may_contain_alias(const SchemaArgument& lhs, const SchemaArgument& rhs, bool b… in may_contain_alias()
201 std::ostream& operator<<(std::ostream& out, const FunctionSchema& schema) { in operator <<()
[all …]
Dbuiltin_function.h15 c10::FunctionSchema schema,
51 const c10::FunctionSchema& getSchema() const override { in getSchema()
59 Function& setSchema(c10::FunctionSchema schema) override { in setSchema()
85 c10::FunctionSchema schema_;
/external/executorch/exir/operator/
Dconvert.py32 from torchgen.model import FunctionSchema, SchemaKind
37 _op_overload_to_schema_cache: Dict[OpOverload, FunctionSchema] = {}
49 _schema_mismatch_map: Dict[OpOverload, Optional[FunctionSchema]] = {}
53 pybind_schema: torch._C.FunctionSchema, argument
54 ) -> Optional[FunctionSchema]:
64 native_schema = FunctionSchema.parse(str(pybind_schema))
85 def _get_overload_schema(op_overload: OpOverload) -> Optional[FunctionSchema]:
97 def get_out_args_from_schema(out_var_schema: FunctionSchema) -> Tuple[str]: argument
131 def schema_to_opoverload(schema: FunctionSchema) -> OpOverload: argument
173 group_by_signature: Dict[str, Dict[SchemaKind, FunctionSchema]] = {}
[all …]
/external/pytorch/torch/csrc/jit/backends/
Dbackend.h12 inline c10::FunctionSchema getIsAvailableSchema() { in getIsAvailableSchema()
15 c10::FunctionSchema preprocessor_schema( in getIsAvailableSchema()
26 inline c10::FunctionSchema getCompileSchema() { in getCompileSchema()
34 c10::FunctionSchema compile_schema( in getCompileSchema()
43 inline c10::FunctionSchema getExecuteSchema() { in getExecuteSchema()
49 return c10::FunctionSchema( in getExecuteSchema()
/external/pytorch/torchgen/
Dnative_function_generation.py21 FunctionSchema,
106 ) -> dict[FunctionSchema, dict[SchemaKind, NativeFunction]]:
108 FunctionSchema, dict[SchemaKind, NativeFunction]
127 def self_to_out_signature(func: FunctionSchema) -> FunctionSchema: argument
135 return FunctionSchema(
160 def functional_to_out_signature(func: FunctionSchema) -> FunctionSchema: argument
169 return FunctionSchema(
182 func: FunctionSchema, argument
234 def mutable_to_out_signature(func: FunctionSchema) -> FunctionSchema: argument
252 return FunctionSchema(
[all …]
/external/pytorch/torch/_library/
Dutils.py97 if isinstance(schema, torch._C.FunctionSchema):
101 from torchgen.model import FunctionSchema
104 schema = FunctionSchema.parse(schema)
105 assert isinstance(schema, FunctionSchema)
181 schema: _C.FunctionSchema, args: Tuple[Any, ...], kwargs: Dict[str, Any] argument
285 def has_kwarg_only_args(schema: _C.FunctionSchema): argument
289 def has_kwarg_only_tensors(schema: _C.FunctionSchema): argument
299 def has_tensor_arg(schema: _C.FunctionSchema) -> bool: argument
310 def get_device_arg_index(schema: _C.FunctionSchema) -> Union[int, None]: argument
/external/pytorch/aten/src/ATen/core/dispatch/
DOperatorEntry.h38 AnnotatedKernel(KernelFunction k, std::unique_ptr<FunctionSchema> s, std::string d) in AnnotatedKernel()
45 std::unique_ptr<FunctionSchema> inferred_function_schema;
55 AnnotatedSchema(FunctionSchema s, std::string d) in AnnotatedSchema()
59 FunctionSchema schema;
79 const FunctionSchema& schema() const { in schema()
103 void registerSchema(FunctionSchema&&, std::string&& debug, std::vector<at::Tag> tags = {});
135 std::unique_ptr<FunctionSchema> inferred_function_schema,
/external/executorch/exir/dialects/edge/op/
Dapi.py18 from torchgen.model import FunctionSchema, SchemaKind
52 native_schema: Optional[FunctionSchema] = _pybind_schema_to_native_schema(
63 schemas: List[torch._C.FunctionSchema] = [
70 native_s: Optional[FunctionSchema] = _pybind_schema_to_native_schema(schema)
/external/pytorch/torch/csrc/jit/frontend/
Dschema_matching.h22 TORCH_API bool isBlockListedSchema(const FunctionSchema& schema);
25 const ::c10::FunctionSchema& schema,
33 const std::vector<const ::c10::FunctionSchema*>& schemas,
45 TORCH_API std::string getFullSchemaName(const ::c10::FunctionSchema& schema);
Dfunction_schema_parser.cpp16 using c10::FunctionSchema;
34 std::variant<OperatorName, FunctionSchema> parseDeclaration() { in parseDeclaration()
98 return FunctionSchema( in parseDeclaration()
131 std::vector<std::variant<OperatorName, FunctionSchema>> parseDeclarations() { in parseDeclarations()
132 std::vector<std::variant<OperatorName, FunctionSchema>> results; in parseDeclarations()
140 std::variant<OperatorName, FunctionSchema> parseExactlyOneDeclaration() { in parseExactlyOneDeclaration()
404 std::variant<OperatorName, FunctionSchema> parseSchemaOrName( in parseSchemaOrName()
416 FunctionSchema parseSchema(const std::string& schema, bool allow_typevars) { in parseSchema()
419 std::holds_alternative<FunctionSchema>(parsed), in parseSchema()
421 return std::get<FunctionSchema>(std::move(parsed)); in parseSchema()
/external/pytorch/torch/
Dlibrary.h305 std::unique_ptr<c10::FunctionSchema> schema_;
320 std::unique_ptr<c10::FunctionSchema> schema);
409 inline c10::FunctionSchema schema(const char* str, c10::AliasAnalysisKind k, bool allow_typevars=fa…
410 c10::FunctionSchema s = torch::jit::parseSchema(str, /*allow_typevars*/allow_typevars);
418 inline c10::FunctionSchema schema(const char* s, bool allow_typevars=false) {
428 inline c10::FunctionSchema&& schema(c10::FunctionSchema&& s) { in schema()
434 inline std::variant<c10::OperatorName, c10::FunctionSchema> constructSchemaOrName( in constructSchemaOrName()
435 c10::FunctionSchema&& s) { in constructSchemaOrName()
438 inline std::variant<c10::OperatorName, c10::FunctionSchema> constructSchemaOrName( in constructSchemaOrName()
442 inline std::variant<c10::OperatorName, c10::FunctionSchema>
[all …]
/external/pytorch/torch/csrc/jit/mobile/
Dfunction.h23 std::optional<c10::FunctionSchema> schema);
50 torch::jit::Function& setSchema(c10::FunctionSchema schema) override;
52 const c10::FunctionSchema& getSchema() const override;
74 std::optional<c10::FunctionSchema> schema_; // (byte-code version 4+)
/external/pytorch/torch/csrc/utils/
Dschema_info.h9 std::pair<c10::FunctionSchema, std::unordered_set<std::string>>;
19 explicit SchemaInfo(c10::FunctionSchema schema) in SchemaInfo()
86 static std::vector<c10::FunctionSchema> getNonDeterministicOps();
109 const c10::FunctionSchema schema_;
/external/pytorch/torchgen/api/
Dfunctionalization.py19 FunctionSchema,
127 def capture_arguments(func: FunctionSchema, *, is_reverse: bool) -> list[Binding]: argument
145 def returns_type(func: FunctionSchema) -> CType: argument
162 def inner_call_index(func: FunctionSchema) -> Binding | None: argument
172 def inner_arguments(func: FunctionSchema, is_reverse: bool) -> list[Binding]: argument
Ddispatcher.py10 FunctionSchema,
35 def name(func: FunctionSchema) -> str: argument
81 def jit_arguments(func: FunctionSchema) -> list[Argument]: argument
119 def arguments(func: FunctionSchema, *, symint: bool = True) -> list[Binding]: argument

123456