37 x_shape =
stack_[0]->OutputShape(x_shape);
41 return stack_[0]->OutputShape(input_shape);
55 ReverseData(input, rev_input);
57 stack_[0]->Forward(debug, *rev_input,
nullptr, scratch, rev_output);
58 ReverseData(*rev_output,
output);
66 ReverseData(fwd_deltas, rev_input);
69 ReverseData(*rev_output, back_deltas);
78 dest->CopyWithXReversal(src);
80 dest->CopyWithYReversal(src);
82 dest->CopyWithXYTranspose(src);
virtual void AddToStack(Network *network)
std::vector< Network * > stack_
void Forward(bool debug, const NetworkIO &input, const TransposedArray *input_transpose, NetworkScratch *scratch, NetworkIO *output) override
TESS_API Reversed(const std::string &name, NetworkType type)
TESS_API void SetNetwork(Network *network)
StaticShape OutputShape(const StaticShape &input_shape) const override
bool Backward(bool debug, const NetworkIO &fwd_deltas, NetworkScratch *scratch, NetworkIO *back_deltas) override
void SetShape(int batch, int height, int width, int depth)
void set_width(int value)
void set_height(int value)