Home
last modified time | relevance | path

Searched refs:partition_types (Results 1 – 7 of 7) sorted by relevance

/external/executorch/backends/cadence/aot/quantizer/
Dpatterns.py50 def partition_types(self) -> list[OpOverload]: member in QuantizationPattern
72 def partition_types(self) -> List[OpOverload]: member in AddmmPattern
105 def partition_types(self) -> List[OpOverload]: member in BmmPattern
126 def partition_types(self) -> List[OpOverload]: member in Conv1dPattern
165 def partition_types(self) -> List[OpOverload]: member in Conv2dPattern
204 def partition_types(self) -> List[OpOverload]: member in LayerNormPattern
239 def partition_types(self) -> List[OpOverload]: member in LinearPattern
278 def partition_types(self) -> List[OpOverload]: member in MatmulPattern
301 def partition_types(self) -> List[OpOverload]: member in ReluBasePattern
323 def partition_types(self) -> List[OpOverload]: member in ReluPattern0
[all …]
Dutils.py221 partition_types: List[Any],
224 for partition_type in partition_types:
Dquantizer.py75 self.pattern.partition_types(),
Dfusion_pass.py338 pattern.partition_types(),
/external/pytorch/torch/ao/quantization/pt2e/
Dgraph_utils.py79 def _valid_type_sequence(partition_types: List[Any]):
81 for partition_type in partition_types:
92 partition_types: List[Any],
96 if not _valid_type_sequence(partition_types):
102 for partition_type in partition_types:
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc46 std::vector<tensorflow::RowPartitionType> partition_types; member
51 if (partition_types.front() == in GetRowPartitionTypeByDimension()
53 return partition_types[dimension + 1]; in GetRowPartitionTypeByDimension()
55 return partition_types[dimension]; in GetRowPartitionTypeByDimension()
64 attributes->partition_types.front(); in GetFirstDimensionSizeT()
145 if (conversion_attributes.partition_types.front() == in GetRowPartitionTensor()
566 attributes->partition_types = in Initialize()
568 if (attributes->partition_types.size() != in Initialize()
574 tensorflow::GetRaggedRank(attributes->partition_types); in Initialize()
Dragged_tensor_to_tensor_tflite_test.cc45 std::vector<std::string> partition_types, in RaggedTensorToTensorOpModel() argument
66 for (const auto& s : partition_types) { in RaggedTensorToTensorOpModel()
71 fbb.Int("num_row_partition_tensors", partition_types.size()); in RaggedTensorToTensorOpModel()