| 106 | } |
| 107 | |
| 108 | static ShapeProvider* make( |
| 109 | ComputingGraph& graph, const TensorShape& shape, DType dtype, |
| 110 | const OperatorNodeConfig& config) { |
| 111 | auto opr = graph.insert_opr( |
| 112 | std::make_unique<ShapeProvider>(graph, shape, dtype, config)); |
| 113 | return &opr->cast_final_safe<ShapeProvider>(); |
| 114 | } |
| 115 | |
| 116 | //! update shape |
| 117 | void shape(const TensorShape& shape) { m_shape = shape; } |
no test coverage detected