| 419 | } |
| 420 | |
| 421 | megdnn::TensorShape JITExecutor::broadcasted_input_shape() const { |
| 422 | megdnn::TensorShapeArray inp_shps; |
| 423 | megdnn::TensorShape brdcast_shp; |
| 424 | auto placeholders = m_internal_graph->placeholders(); |
| 425 | for (auto ph : placeholders) { |
| 426 | if (!ph->is_host_value_shape_input()) { |
| 427 | inp_shps.push_back(input(ph->input_id())->shape()); |
| 428 | } |
| 429 | } |
| 430 | megdnn::Elemwise::deduce_shape(inp_shps, brdcast_shp); |
| 431 | return brdcast_shp; |
| 432 | } |
| 433 | |
| 434 | #if MGB_ENABLE_GRAD |
| 435 | namespace { |
no test coverage detected