Searched refs:KernelKey (Results 1 – 5 of 5) sorted by relevance
| /external/executorch/runtime/kernel/test/ |
| D | operator_registry_test.cpp | 29 using executorch::runtime::KernelKey;
62 constexpr int BUF_SIZE = KernelKey::MAX_SIZE;
67 KernelKey long_contiguous = KernelKey(buf_long_contiguous); in TEST_F()
69 KernelKey long_key_1 = KernelKey(long_contiguous); in TEST_F()
71 KernelKey long_key_2 = KernelKey(long_contiguous); in TEST_F()
77 KernelKey float_key = KernelKey(buf_float_contiguous); in TEST_F()
83 KernelKey long_key_3 = KernelKey(buf_channel_first); in TEST_F()
91 KernelKey key = KernelKey(buf_long_contiguous); in TEST_F()
131 KernelKey key_1 = KernelKey(buf_long_contiguous); in TEST_F()
135 KernelKey key_2 = KernelKey(buf_float_contiguous); in TEST_F()
[all …]
|
| /external/executorch/runtime/kernel/ |
| D | operator_registry.h | 123 struct KernelKey { struct
125 KernelKey() : is_fallback_(true) {} in KernelKey() function
127 /* implicit */ KernelKey(const char* kernel_key_data) in KernelKey() function
132 bool operator==(const KernelKey& other) const { argument
136 bool operator!=(const KernelKey& other) const {
140 bool equals(const KernelKey& other) const { in equals() argument
159 friend std::ostream& operator<<(std::ostream& os, const KernelKey& key) { argument
181 KernelKey kernel_key_;
190 explicit Kernel(const char* name, KernelKey key, OpFunction func) in Kernel()
249 using ::executorch::runtime::KernelKey;
|
| D | operator_registry.cpp | 167 char buf[KernelKey::MAX_SIZE] = {0}; in get_op_function_from_registry()
169 KernelKey kernel_key = KernelKey(buf); in get_op_function_from_registry()
|
| /external/executorch/runtime/executor/test/ |
| D | kernel_resolution_test.cpp | 33 using executorch::runtime::KernelKey;
106 KernelKey key = KernelKey("v1/6;0,1|6;0,1|6;0,1|6;0,1"); in TEST_F()
|
| D | kernel_integration_test.cpp | 35 using executorch::runtime::KernelKey;
110 KernelKey key = in register_singleton()
111 executorch::runtime::KernelKey("v1/6;0,1|6;0,1|6;0,1|6;0,1"); in register_singleton()
|