Searched refs:inferred_function_schema (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | OperatorEntry.cpp | 85 if (j.inferred_function_schema != nullptr) { in registerSchema()
86 checkSchema(name_, schema, debug, j.kernel, *j.inferred_function_schema, j.debug); in registerSchema()
109 std::unique_ptr<FunctionSchema> inferred_function_schema, in registerKernel() argument
138 if (schema_ && inferred_function_schema) { in registerKernel()
139 checkSchema(name_, schema_->schema, schema_->debug, kernel, *inferred_function_schema, debug); in registerKernel()
168 k[0].inferred_function_schema = std::move(inferred_function_schema); in registerKernel()
171 k.emplace_front(std::move(kernel), std::move(inferred_function_schema), std::move(debug)); in registerKernel()
615 << (jt.inferred_function_schema ? toString(*jt.inferred_function_schema) : "(none)") in registerKernel()
|
| H A D | OperatorEntry.h | 40 , inferred_function_schema(std::move(s)) in AnnotatedKernel()
45 std::unique_ptr<FunctionSchema> inferred_function_schema; member
135 std::unique_ptr<FunctionSchema> inferred_function_schema,
|
| H A D | Dispatcher.cpp | 344 std::unique_ptr<FunctionSchema> inferred_function_schema, in registerImpl() argument
356 std::move(inferred_function_schema), in registerImpl()
|
| H A D | Dispatcher.h | 224 …l::CppSignature> cpp_signature, std::unique_ptr<FunctionSchema> inferred_function_schema, std::str…
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration.cpp | 76 if (nullptr != kernel.inferred_function_schema.get()) { in inferSchemaFromKernels_()
78 inferred_schema = *kernel.inferred_function_schema; in inferSchemaFromKernels_()
136 std::move(kernel.inferred_function_schema), in registerOp_()
|
| H A D | op_registration.h | 402 …impl::CppSignature> cpp_signature, std::unique_ptr<FunctionSchema>&& inferred_function_schema) && { in kernel() argument
407 config.inferred_function_schema = std::move(inferred_function_schema); in kernel()
425 , inferred_function_schema(nullptr) in KernelRegistrationConfig()
431 std::unique_ptr<FunctionSchema> inferred_function_schema; member
|