broadcast src to broadcasted shape of dst_shape_var
| 39 | |
| 40 | //! broadcast src to broadcasted shape of dst_shape_var |
| 41 | SymbolVar broadcast_tshp(SymbolVar src, const SymbolVarArrayView& dst_shape_var) { |
| 42 | auto dtype = src.dtype(); |
| 43 | for (auto i : dst_shape_var) |
| 44 | dtype = dtype_promotion(dtype, i.dtype()); |
| 45 | src = opr::TypeCvt::make(src, dtype); |
| 46 | return opr::Broadcast::make( |
| 47 | src, opr::GetVarShape::make(VarNodeArrayView{dst_shape_var})); |
| 48 | } |
| 49 | |
| 50 | //! broadcast to ensure returned value shape is compatible with inp |
| 51 | static inline SymbolVar broadcast_ensure(SymbolVar value, SymbolVar inp) { |
no test coverage detected