38 x_shape =
stack_[0]->OutputShape(x_shape);
43 return stack_[0]->OutputShape(input_shape);
58 ReverseData(input, rev_input);
60 stack_[0]->Forward(debug, *rev_input,
nullptr, scratch, rev_output);
61 ReverseData(*rev_output, output);
70 ReverseData(fwd_deltas, rev_input);
73 ReverseData(*rev_output, back_deltas);
82 dest->CopyWithXReversal(src);
84 dest->CopyWithYReversal(src);
86 dest->CopyWithXYTranspose(src);
void SetShape(int batch, int height, int width, int depth)
Reversed(const STRING &name, NetworkType type)
bool Backward(bool debug, const NetworkIO &fwd_deltas, NetworkScratch *scratch, NetworkIO *back_deltas) override
void set_width(int value)
void Forward(bool debug, const NetworkIO &input, const TransposedArray *input_transpose, NetworkScratch *scratch, NetworkIO *output) override
void SetNetwork(Network *network)
PointerVector< Network > stack_
void set_height(int value)
virtual void AddToStack(Network *network)
StaticShape OutputShape(const StaticShape &input_shape) const override