Searched defs:in_ops (Results 1 – 8 of 8) sorted by relevance
38 void TensorRTOp::set_in_ops(const std::vector<TensorRTOp *> &in_ops) { this->in_ops_ = in_ops; } in set_in_ops()42 const std::vector<TensorRTOp *> &TensorRTOp::in_ops() const { return this->in_ops_; } in in_ops() function in mindspore::lite::TensorRTOp
145 std::vector<T *> in_ops; in FindPreOps() local172 auto in_ops = FindPreOps(op, all_ops); in FindPreNextOps() local181 … std::vector<T *> *in_ops, std::vector<T *> *out_ops, const std::vector<T *> &all_ops) { in GetGraphInOutOps()
94 void set_in_ops(const std::vector<NPUOp *> &in_ops) { this->in_ops_ = in_ops; } in set_in_ops()98 const std::vector<NPUOp *> &in_ops() const { return this->in_ops_; } in in_ops() function
51 void NPUPassUtils::UpdateOp(NPUOp *op, const std::vector<NPUOp *> &in_ops, const std::vector<NPUOp … in UpdateOp()179 auto in_ops = op->in_ops(); in OpInputFromOp() local
123 std::vector<NPUOp *> in_ops; in InsertNode() local
100 std::vector<NPUOp *> in_ops; in FindPreOps() local125 auto in_ops = FindPreOps(op); in FindPreNextOps() local
76 … const std::vector<mindspore::MSTensor> &out_tensors, const std::vector<NPUOp *> &in_ops) { in Run()
104 std::vector<NPUOp *> in_ops; in FindPreOps() local