Retain TensorFlow NodeDef in Toco Operator. If an op is supported by Toco but not supported by TFLite, TFLite exporter will use the retained NodeDef to populate a Flex op when Flex mode is enabled. This can't be easily applied to all operations, because a TensorFlow node may become multiple Toco operators. Thus we need to call this function in operator conversion functions one by one whenever fe
| 559 | // TODO(b/117327937): Implement all Toco-supported ops in TFLite, and remove |
| 560 | // this function. |
| 561 | void RetainTensorFlowNodeDef(const NodeDef& node, Operator* op) { |
| 562 | node.SerializeToString(&op->tensorflow_node_def); |
| 563 | } |
| 564 | |
| 565 | void GetOutputNamesFromNodeDef(const NodeDef& node, |
| 566 | const tensorflow::OpDef& op_def, |
no test coverage detected