Home
last modified time | relevance | path

Searched refs:model_fd (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/tools/test_generator/
Dcts_generator.py98 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 …]
Dvts_generator.py299 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,
Dspec_visualizer.py403 DumpModel=None, model_fd=None,