MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / PadOpConstModel

Class PadOpConstModel

tensorflow/lite/kernels/pad_test.cc:113–128  ·  view source on GitHub ↗

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();

Source from the content-addressed store, hash-verified

111// m.SetInput(input_data);
112// m.Invoke();
113class 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.
131template <typename RegularInputOuput>

Callers 1

TESTFunction · 0.70

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.56