Searched refs:DataLayoutConverter (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/tools/test_generator/tests/P_naming/ |
| D | conv_float.mod.py | 41 nchw = DataLayoutConverter("NCHW", name="nchw_layout").Identify([i1, f1, output], [layout])
|
| /frameworks/ml/nn/tools/test_generator/tests/P_vts_naming/ |
| D | conv_float.mod.py | 41 nchw = DataLayoutConverter("NCHW", name="nchw_layout").Identify([i1, f1, output], [layout])
|
| /frameworks/ml/nn/tools/test_generator/ |
| D | README.md | 117 - DataLayoutConverter
135 #### DataLayoutConverter subsubsection
140 converter = DataLayoutConverter(target_data_layout, name="variation_name").Identify(
264 # DataLayoutConverter
265 example.AddVariations(DataLayoutConverter("nchw").Identify(op_list))
|
| D | spec_visualizer.py | 39 from test_generator import DataLayoutConverter
|
| D | test_generator.py | 824 class DataLayoutConverter(ModelVariation, ImplicitVariation): class
829 assert DataLayoutConverter.IsCompatible(self.targetLayout)
1110 var = DataLayoutConverter("nchw").Identify(args)
1369 DataLayoutConverter=DataLayoutConverter,
|