| 4 | |
| 5 | namespace TEngine { |
| 6 | bool Ceil::InferShape(const std::vector<TEngine::TShape>& ishape, std::vector<TEngine::TShape>& oshape, int layout) |
| 7 | { |
| 8 | const TShape& input = ishape[0]; |
| 9 | const std::vector<int>& in_dim = input.GetDim(); |
| 10 | |
| 11 | TShape shape; |
| 12 | |
| 13 | shape.SetDim(in_dim); |
| 14 | shape.SetDataLayout(input.GetDataLayout()); |
| 15 | |
| 16 | oshape[0] = shape; |
| 17 | |
| 18 | return true; |
| 19 | } |
| 20 | void Ceil::SetSchema(void) |
| 21 | { |
| 22 | Input({"input:float32"}) |
nothing calls this directly
no test coverage detected