| 87 | }; |
| 88 | |
| 89 | void SimpleOpModel::Init( |
| 90 | const std::function<TfLiteRegistration*()>& registration) { |
| 91 | inputs_[0] = AddInput({TensorType_INT32, {1}}); |
| 92 | inputs_[1] = AddInput({TensorType_INT32, {1}}); |
| 93 | output_ = AddOutput({TensorType_INT32, {}}); |
| 94 | SetCustomOp(kOpName, {}, registration); |
| 95 | BuildInterpreter({GetShape(inputs_[0]), GetShape(inputs_[1])}); |
| 96 | } |
| 97 | |
| 98 | TEST(ProfileSummarizerTest, Empty) { |
| 99 | ProfileSummarizer summarizer; |