Searched refs:KernelTypes (Results 1 – 11 of 11) sorted by relevance
27 case executorch_flatbuffer::KernelTypes::Null: { in get_tag()30 case executorch_flatbuffer::KernelTypes::Int: { in get_tag()33 case executorch_flatbuffer::KernelTypes::Double: { in get_tag()36 case executorch_flatbuffer::KernelTypes::Bool: { in get_tag()39 case executorch_flatbuffer::KernelTypes::String: { in get_tag()42 case executorch_flatbuffer::KernelTypes::Tensor: { in get_tag()
310 executorch_flatbuffer::KernelTypes::Null || in parse_values()317 case executorch_flatbuffer::KernelTypes::Null: { in parse_values()323 case executorch_flatbuffer::KernelTypes::Int: { in parse_values()326 case executorch_flatbuffer::KernelTypes::Double: { in parse_values()330 case executorch_flatbuffer::KernelTypes::Bool: { in parse_values()334 case executorch_flatbuffer::KernelTypes::IntList: { in parse_values()354 case executorch_flatbuffer::KernelTypes::BoolList: { in parse_values()368 case executorch_flatbuffer::KernelTypes::DoubleList: { in parse_values()375 case executorch_flatbuffer::KernelTypes::String: { in parse_values()384 case executorch_flatbuffer::KernelTypes::Tensor: { in parse_values()[all …]
94 executorch_flatbuffer::KernelTypes::Tensor) { in test_module_add()105 executorch_flatbuffer::KernelTypes::Double) { in test_module_add()146 executorch_flatbuffer::KernelTypes::Tensor && in TEST_F()
26 supported_program_type_table: Dict[Type[core_schema.KernelTypes], ConfigValue] = {175 ) -> core_schema.KernelTypes:184 ) -> core_schema.KernelTypes:
29 using KT = KernelTypes<Element>;
209 using KT = KernelTypes<Element>;
122 struct KernelTypes { struct
30 KernelTypes,73 def comp_types(val: KernelTypes, input_val: ValueType) -> Optional[bool]: # noqa argument
139 KernelTypes = Union[ variable
189 union KernelTypes {205 val: KernelTypes;
46 KernelTypes,138 self.assertTrue(type(value.val) in KernelTypes.__args__)