30 std::vector<mir::Operation::Output *> inputs = context->
getNodeInputs(onnx_node);
32 const auto &input_shape = inputs[0]->getShape();
33 int size = input_shape.rank();
35 std::vector<int64_t> data(
static_cast<std::size_t
>(
size));
36 for (
int i = 0; i <
size; i++)
38 data[i] = input_shape.dim(i);
41 auto result = createOp<mir::ops::ConstantOp>(graph, tensor)->getOutput(0);
void setNodeOutputs(const onnx::NodeProto &onnx_node, const std::vector< mir::Operation::Output * > &outputs)
std::vector< mir::Operation::Output * > getNodeInputs(const onnx::NodeProto &onnx_node) const
mir::Graph * getGraph() const