Searched refs:DataLayoutConverter (Results 1 – 5 of 5) sorted by relevance
/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/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/ |
D | README.md | 117 - DataLayoutConverter 135 #### DataLayoutConverter subsubsection 140 converter = DataLayoutConverter(target_data_layout, name="variation_name").Identify( 269 # DataLayoutConverter 270 example.AddVariations(DataLayoutConverter("nchw").Identify(op_list))
|
D | spec_visualizer.py | 39 from test_generator import DataLayoutConverter
|
D | test_generator.py | 863 class DataLayoutConverter(ModelVariation, ImplicitVariation): class 868 assert DataLayoutConverter.IsCompatible(self.targetLayout) 1148 var = DataLayoutConverter("nchw").Identify(args) 1411 DataLayoutConverter=DataLayoutConverter,
|