|
|
|
@ -99,8 +99,7 @@ public:
|
|
|
|
|
ConvFunctionBase::init(config);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void check(const BufferArgs& inputs,
|
|
|
|
|
const BufferArgs& outputs) override {
|
|
|
|
|
void check(const BufferArgs& inputs, const BufferArgs& outputs) override {
|
|
|
|
|
const TensorShape& input = inputs[0].shape();
|
|
|
|
|
const TensorShape& filter = inputs[1].shape();
|
|
|
|
|
const TensorShape& output = outputs[0].shape();
|
|
|
|
@ -162,8 +161,7 @@ public:
|
|
|
|
|
ConvFunctionBase::init(config);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void check(const BufferArgs& inputs,
|
|
|
|
|
const BufferArgs& outputs) override {
|
|
|
|
|
void check(const BufferArgs& inputs, const BufferArgs& outputs) override {
|
|
|
|
|
const TensorShape& output = inputs[0].shape();
|
|
|
|
|
const TensorShape& filter = inputs[1].shape();
|
|
|
|
|
const TensorShape& input = outputs[0].shape();
|
|
|
|
@ -225,8 +223,7 @@ public:
|
|
|
|
|
ConvFunctionBase::init(config);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
virtual void check(const BufferArgs& inputs,
|
|
|
|
|
const BufferArgs& outputs) override {
|
|
|
|
|
void check(const BufferArgs& inputs, const BufferArgs& outputs) override {
|
|
|
|
|
const TensorShape& output = inputs[0].shape();
|
|
|
|
|
const TensorShape& input = inputs[1].shape();
|
|
|
|
|
const TensorShape& filter = outputs[0].shape();
|
|
|
|
|