|
|
@ -74,9 +74,9 @@ public:
|
|
|
|
|
|
|
|
|
|
|
|
virtual void check(const BufferArgs& inputs,
|
|
|
|
virtual void check(const BufferArgs& inputs,
|
|
|
|
const BufferArgs& outputs) override {
|
|
|
|
const BufferArgs& outputs) override {
|
|
|
|
const TensorShape& output = inputs[0].shape();
|
|
|
|
const TensorShape& input = inputs[0].shape();
|
|
|
|
const TensorShape& filter = inputs[1].shape();
|
|
|
|
const TensorShape& filter = inputs[1].shape();
|
|
|
|
const TensorShape& input = outputs[0].shape();
|
|
|
|
const TensorShape& output = outputs[0].shape();
|
|
|
|
checkShape(input, filter, output);
|
|
|
|
checkShape(input, filter, output);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|