Searched refs:OperatorRegistrationTest (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/core/op_registration/ |
| D | op_registration_test.cpp | 51 TEST(OperatorRegistrationTest, whenRegisteringWithSchemaBeforeKernelInOptionsObject_thenCanBeCalled… in TEST() argument
62 TEST(OperatorRegistrationTest, whenRegisteringWithSchemaAfterKernelInOptionsObject_thenCanBeCalled)… in TEST() argument
73 TEST(OperatorRegistrationTest, whenRegisteringWithNameBeforeKernelInOptionsObject_thenCanBeCalled) { in TEST() argument
84 TEST(OperatorRegistrationTest, whenRegisteringWithNameAfterKernelInOptionsObject_thenCanBeCalled) { in TEST() argument
95 TEST(OperatorRegistrationTest, whenRegisteringWithoutSchema_thenFails) { in TEST() argument
101 TEST(OperatorRegistrationTest, whenCallingOpWithWrongDispatchKey_thenFails) { in TEST() argument
112 TEST(OperatorRegistrationTest, givenOpWithCatchallKernel_whenCallingOp_thenCallsCatchallKernel) { in TEST() argument
141 TEST(OperatorRegistrationTest, givenOpWithDispatchedKernelOutOfScope_whenRegisteringCatchallKernelA… in TEST() argument
174 TEST(OperatorRegistrationTest, givenOpWithCatchallKernelOutOfScope_whenRegisteringDispatchedKernelA… in TEST() argument
189 TEST(OperatorRegistrationTest, givenOpWithoutKernels_whenRegisteringWithSchema_thenOnlyRegistersSch… in TEST() argument
[all …]
|
| /external/pytorch/test/edge/ |
| D | test_operator_registration.cpp | 10 TEST(OperatorRegistrationTest, Add) { in TEST() argument
32 TEST(OperatorRegistrationTest, CustomAdd3) { in TEST() argument
|