Searched refs:model_fd (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/tools/test_generator/ |
D | cts_generator.py | 98 def InitializeFiles(model_fd, example_fd, test_fd): argument 111 print(fileHeader.format(spec_file=specFileBase), file=model_fd) 120 def DumpCtsIsIgnored(model, model_fd): argument 127 is_ignored_name=str(model.isIgnoredFunctionName)), file=model_fd) 130 def DumpCtsModel(model, model_fd): argument 134 print("void %s(Model *model) {"%(model.createFunctionName), file=model_fd) 149 IndentedPrint(typeDef, file=model_fd) 152 print(" // Phase 1, operands", file=model_fd) 154 IndentedPrint("auto %s = model->addOperand(&%s);"%(op, op.type), file=model_fd) 157 print(" // Phase 2, operations", file=model_fd) [all …]
|
D | vts_generator.py | 299 def InitializeFiles(model_fd, example_fd, test_fd): argument 312 print(fileHeader.format(spec_file=specFileBase), file=model_fd) 327 with SmartOpen(tg.FileNames.modelFile) as model_fd, \ 330 InitializeFiles(model_fd, example_fd, test_fd) 332 DumpModel=generate_vts, model_fd=model_fd,
|
D | spec_visualizer.py | 403 DumpModel=None, model_fd=None,
|