Tests case where paddings is a const tensor. Example usage is as follows: PadOpDynamicModel m(input_shape, paddings_shape, paddings_data); m.SetInput(input_data); m.Invoke();
| 111 | // m.SetInput(input_data); |
| 112 | // m.Invoke(); |
| 113 | class PadOpConstModel : public PadOpModel<float> { |
| 114 | public: |
| 115 | PadOpConstModel(const TensorData& input, |
| 116 | std::initializer_list<int> paddings_shape, |
| 117 | std::initializer_list<int> paddings, |
| 118 | const TensorData& output) { |
| 119 | this->input_ = AddInput(input); |
| 120 | paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape); |
| 121 | constant_values_ = AddNullInput(); |
| 122 | output_ = AddOutput(output); |
| 123 | |
| 124 | SetBuiltinOp(BuiltinOperator_PAD, BuiltinOptions_PadOptions, |
| 125 | CreatePadOptions(builder_).Union()); |
| 126 | BuildInterpreter({input.shape}); |
| 127 | } |
| 128 | }; |
| 129 | |
| 130 | // Test case where paddings is a non-const tensor. |
| 131 | template <typename RegularInputOuput> |