Home
last modified time | relevance | path

Searched defs:out_tensors (Results 1 – 25 of 197) sorted by relevance

12345678

/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dnpu_op.h43 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in NPUOp()
53 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
58 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
63 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
69 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
88 …void set_outputs(const std::vector<mindspore::MSTensor> &out_tensors) { this->outputs_ = out_tenso… in set_outputs()
122 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) { in GetNPUOp()
Dreshape_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
36 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
46 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
54 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dactivation_npu.cc21 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
38 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
63 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
70 …onst std::vector<mindspore::MSTensor> &in_tensors, const std::vector<mindspore::MSTensor> &out_ten… in SetNPUInputs()
Dtranspose_npu.h27const std::vector<mindspore::MSTensor> &out_tensors, std::vector<int> perm, std::string name) in TransposeNPUOp()
34 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in TransposeNPUOp()
43 const std::vector<mindspore::MSTensor> &out_tensors) override { in Init()
Dcast_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
38 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
54 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dgather_npu.cc24 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
40 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
51 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dreduce_npu.cc21 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
40 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
62 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dunsqueeze_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
32 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
60 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Darithmetic_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
52 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
145 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
204 …onst std::vector<mindspore::MSTensor> &in_tensors, const std::vector<mindspore::MSTensor> &out_ten… in SetNPUInputs()
Dtile_npu.cc23 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
35 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
45 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dcrop_and_resize_npu.cc26 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
47 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
72 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dmatmul_npu.cc25 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
35 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
55 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dconvolution_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
66 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
95 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
116 …onst std::vector<mindspore::MSTensor> &in_tensors, const std::vector<mindspore::MSTensor> &out_ten… in SetNPUInputs()
160 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) { in GetNPUConvOp()
Dconcat_npu.cc22 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
38 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
50 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dresize_npu.cc26 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
47 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
109 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dstrided_slice_npu.cc23 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
47 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
67 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dsqueeze_npu.h27 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in SqueezeNPUOp()
33 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Dexpand_dims_npu.h28 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in ExpandDimsNPUOp()
33 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Dsoftmax_npu.h28 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in SoftmaxNPUOp()
34 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Darithmetic_self_npu.h28 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in ArithmeticSelfNPUOp()
34 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Dslice_npu.h28 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in SliceNPUOp()
34 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Dmax_pooling_npu.cc21 const std::vector<mindspore::MSTensor> &out_tensors) { in IsSupport()
68 const std::vector<mindspore::MSTensor> &out_tensors) { in Init()
96 const std::vector<mindspore::MSTensor> &out_tensors, in SetNPUInputs()
Dconcat_npu.h29 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in ConcatNPUOp()
35 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
Deltwise_npu.h28 const std::vector<mindspore::MSTensor> &out_tensors, std::string name) in EltwiseNPUOp()
34 const std::vector<mindspore::MSTensor> &out_tensors) override { in IsSupport()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Drelux_int8_coder.h31 ReluxInt8Coder(const std::vector<Tensor *> &in_tensors, const std::vector<Tensor *> &out_tensors, in ReluxInt8Coder()
50 ReluInt8Coder(const std::vector<Tensor *> &in_tensors, const std::vector<Tensor *> &out_tensors, in ReluInt8Coder()
66 Relu6Int8Coder(const std::vector<Tensor *> &in_tensors, const std::vector<Tensor *> &out_tensors, in Relu6Int8Coder()

12345678