File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 33#include " tensor.hpp"
44#include < iostream>
55
6- namespace infinicore {
7- std::ostream &operator <<(std::ostream &out, const Tensor &tensor);
8- }
9-
106namespace infinicore ::print_options {
117
128/* *
Original file line number Diff line number Diff line change 66#include < type_traits>
77#include < unordered_map>
88#include < vector>
9- #include < spdlog/spdlog.h>
109
1110namespace infinicore ::nn {
1211class Module {
Original file line number Diff line number Diff line change 44#include " ../tensor.hpp"
55#include " module.hpp"
66#include < memory>
7+ #include < cmath>
78
89namespace infinicore ::nn {
910
Original file line number Diff line number Diff line change @@ -92,6 +92,7 @@ class Tensor {
9292 friend class TensorImpl ;
9393
9494 void resume_from_blob_ () const ;
95+ friend std::ostream &operator <<(std::ostream &out, const Tensor &tensor);
9596};
9697
9798class TensorImpl : public std ::enable_shared_from_this<TensorImpl> {
You can’t perform that action at this time.
0 commit comments