Searched refs:flex_ops (Results 1 – 4 of 4) sorted by relevance
22 std::string OpListToJSONString(const OpKernelSet& flex_ops) { in OpListToJSONString() argument24 for (const OpKernel& op : flex_ops) { in OpListToJSONString()30 void AddFlexOpsFromModel(const tflite::Model* model, OpKernelSet* flex_ops) { in AddFlexOpsFromModel() argument54 flex_ops->insert({tf_op_name, ""}); in AddFlexOpsFromModel()
35 tflite::flex::OpKernelSet flex_ops; in main() local46 tflite::flex::AddFlexOpsFromModel(model, &flex_ops); in main()48 std::cout << tflite::flex::OpListToJSONString(flex_ops); in main()
36 std::string OpListToJSONString(const OpKernelSet& flex_ops) { in OpListToJSONString() argument38 for (const OpKernel& op : flex_ops) { in OpListToJSONString()78 void AddFlexOpsFromModel(const tflite::Model* model, OpKernelSet* flex_ops) { in AddFlexOpsFromModel() argument122 flex_ops->insert({tf_op_name, kernel_class}); in AddFlexOpsFromModel()
46 void AddFlexOpsFromModel(const tflite::Model* model, OpKernelSet* flex_ops);50 std::string OpListToJSONString(const OpKernelSet& flex_ops);