| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_arithmetic_parser.cc | 54 std::vector<std::string> *inputs, int *output_size) { in Parse()
68 std::vector<std::string> *inputs, int *output_size) { in Parse()
82 std::vector<std::string> *inputs, int *output_size) { in Parse()
96 std::vector<std::string> *inputs, int *output_size) { in Parse()
110 std::vector<std::string> *inputs, int *output_size) { in Parse()
124 std::vector<std::string> *inputs, int *output_size) { in Parse()
138 std::vector<std::string> *inputs, int *output_size) { in Parse()
152 std::vector<std::string> *inputs, int *output_size) { in Parse()
166 std::vector<std::string> *inputs, int *output_size) { in Parse()
180 std::vector<std::string> *inputs, int *output_size) { in Parse()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/ |
| D | op_node.h | 65 virtual void CheckShape(const NodePtrList &inputs, const DAttrs &attrs) {} in CheckShape()
69 …virtual DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) { return inputs[0]->shap… in InferShape()
70 …virtual TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) { return inputs[0]->type;… in InferType()
71 …virtual DFormat InferFormat(const NodePtrList &inputs, const DAttrs &attrs) { return inputs[0]->fo… in InferFormat()
105 …DShape InferShape(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[2]->sha… in InferShape()
106 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[2]->type… in InferType()
107 …DFormat InferFormat(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[2]->f… in InferFormat()
117 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return inputs[1]->type… in InferType()
126 …TypeId InferType(const NodePtrList &inputs, const DAttrs &attrs) override { return TypeId::kNumber… in InferType()
166 DFormat InferFormat(const NodePtrList &inputs, const DAttrs &attrs) override { in InferFormat()
[all …]
|
| D | op_node.cc | 41 void PrimOp::Check(const NodePtrList &inputs, const DAttrs &attrs) { in Check()
48 void PrimOp::CheckType(const NodePtrList &inputs, const DAttrs &attrs) { in CheckType()
58 void PrimOp::CheckFormat(const NodePtrList &inputs, const DAttrs &attrs) { in CheckFormat()
74 NodeBase PrimOp::Infer(const NodePtrList &inputs, const DAttrs &attrs) { in Infer()
110 tensor::TensorPtr CalcByOperator(const NodePtrList &inputs, const std::string &op, TypeId tid) { in CalcByOperator()
133 NodePtr PrimOp::InferValue(const NodePtrList &inputs, const DAttrs &attrs, const std::string &op) { in InferValue()
223 DShape BroadcastShape(const NodePtrList &inputs, bool to_nz = false) { in BroadcastShape()
258 DShape ElemwiseOp::InferShape(const NodePtrList &inputs, const DAttrs &) { in InferShape()
273 DFormat ElemwiseOp::InferFormat(const NodePtrList &inputs, const DAttrs &attrs) { in InferFormat()
278 NodeBase ElemwiseOp::Infer(const NodePtrList &inputs, const DAttrs &attrs) { in Infer()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.h | 125 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost()
135 …double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou… in GetComputationCost()
153 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost()
161 …double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou… in GetComputationCost()
267 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost()
275 …double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou… in GetComputationCost()
324 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost()
332 …double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou… in GetComputationCost()
352 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost()
360 …double GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &ou… in GetComputationCost()
[all …]
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | bprop_util.py | 43 def construct(self, *inputs):
56 def bprop(func, *inputs, grads_wrt_outputs=None, wrt: list = None, params: list = None):
92 def func_pynative(*inputs):
94 def _func_pynative(*inputs):
|
| /third_party/mindspore/tests/ut/cpp/parallel/ops_info/ |
| D | matmul_info_test.cc | 97 Strategys inputs = {{2, 4, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
108 Strategys inputs = {{2, 4, 8, 8}, {2, 4, 8, 2}}; in TEST_F() local
120 Strategys inputs = {{2, 4, 8, 16}, {1, 16}}; in TEST_F() local
132 Strategys inputs = {{2, 4, 8, 8}, {2, 8}}; in TEST_F() local
144 Strategys inputs = {{8, 16}, {2, 4, 1, 16}}; in TEST_F() local
156 Strategys inputs = {{8, 8}, {2, 4, 2, 8}}; in TEST_F() local
171 std::vector<TensorInfo> inputs = matmul1->inputs_tensor_info(); in TEST_F() local
197 std::vector<TensorInfo> inputs = matmul2->inputs_tensor_info(); in TEST_F() local
223 std::vector<TensorInfo> inputs = matmul3->inputs_tensor_info(); in TEST_F() local
248 std::vector<TensorInfo> inputs = matmul1->inputs_tensor_info(); in TEST_F() local
[all …]
|
| D | softmax_info_test.cc | 71 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
86 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F() local
107 std::vector<TensorInfo> inputs = softmax->inputs_tensor_info(); in TEST_F() local
124 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
135 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
148 Strategys inputs = {{2, 2, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
157 Strategys inputs = {{2, 4, 8}}; in TEST_F() local
166 Strategys inputs = {{2, 4, 8, 16}}; in TEST_F() local
175 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
184 Strategys inputs = {{2, 4, 1, 100}}; in TEST_F() local
|
| D | l2_normalize_info_test.cc | 67 Strategys inputs = {{4, 1, 8}}; in TEST_F() local
82 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F() local
103 std::vector<TensorInfo> inputs = norm->inputs_tensor_info(); in TEST_F() local
120 Strategys inputs = {{4, 1, 8}}; in TEST_F() local
131 Strategys inputs = {{4, 1, 8}}; in TEST_F() local
143 Strategys inputs = {{4, 1, 8}, {4, 1, 8}}; in TEST_F() local
151 Strategys inputs = {{4, 2, 3}}; in TEST_F() local
159 Strategys inputs = {{4, 2, 3, 4}}; in TEST_F() local
167 Strategys inputs = {{4, 1, 8}}; in TEST_F() local
175 Strategys inputs = {{2, 1, 8}}; in TEST_F() local
|
| D | log_softmax_info_test.cc | 67 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
82 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F() local
103 std::vector<TensorInfo> inputs = log_softmax->inputs_tensor_info(); in TEST_F() local
120 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
131 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
144 Strategys inputs = {{2, 2, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
153 Strategys inputs = {{2, 4, 8}}; in TEST_F() local
162 Strategys inputs = {{2, 4, 8, 16}}; in TEST_F() local
170 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
|
| D | reshape_test.cc | 71 Strategys inputs = {{4, 1, 1, 1}}; in TEST_F() local
82 Strategys inputs = {{32, 1, 1, 1}}; in TEST_F() local
97 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F() local
118 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F() local
139 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F() local
160 std::vector<TensorInfo> inputs = reshape->inputs_tensor_info(); in TEST_F() local
177 Strategys inputs = {{4, 1, 1, 1}}; in TEST_F() local
188 Strategys inputs = {{4, 1, 1, 1}}; in TEST_F() local
200 Strategys inputs = {{1, 4, 8}}; in TEST_F() local
208 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
[all …]
|
| D | reduce_method_test.cc | 71 Strategys inputs = {{4, 8, 1}}; in TEST_F() local
86 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F() local
107 std::vector<TensorInfo> inputs = reduce_sum->inputs_tensor_info(); in TEST_F() local
124 Strategys inputs = {{4, 8, 1}}; in TEST_F() local
135 Strategys inputs = {{4, 4, 2}}; in TEST_F() local
159 Strategys inputs = {{4, 8, 1}}; in TEST_F() local
171 Strategys inputs = {{4, 4, 1}}; in TEST_F() local
190 Strategys inputs = {{2, 2, 8, 16}}; in TEST_F() local
198 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
206 Strategys inputs = {{4, 4, 2}}; in TEST_F() local
[all …]
|
| D | gelu_info_test.cc | 66 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
81 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F() local
102 std::vector<TensorInfo> inputs = gelu->inputs_tensor_info(); in TEST_F() local
119 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
130 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
143 Strategys inputs = {{2, 2, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
152 Strategys inputs = {{2, 4, 8}}; in TEST_F() local
161 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
|
| D | pow_info_test.cc | 66 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
81 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F() local
102 std::vector<TensorInfo> inputs = pow->inputs_tensor_info(); in TEST_F() local
119 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
130 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
142 Strategys inputs = {{2, 2, 8}, {2, 4, 8}}; in TEST_F() local
150 Strategys inputs = {{2, 4, 8, 16}, {2, 4, 8, 16}}; in TEST_F() local
158 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
|
| D | tanh_info_test.cc | 66 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
81 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F() local
102 std::vector<TensorInfo> inputs = tanh->inputs_tensor_info(); in TEST_F() local
119 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
130 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
143 Strategys inputs = {{2, 2, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
152 Strategys inputs = {{2, 4, 8}}; in TEST_F() local
161 Strategys inputs = {{2, 4, 1, 16}}; in TEST_F() local
|
| D | transpose_test.cc | 71 Strategys inputs = {{4, 8}}; in TEST_F() local
82 Strategys inputs = {{4, 1}}; in TEST_F() local
97 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F() local
118 std::vector<TensorInfo> inputs = transpose->inputs_tensor_info(); in TEST_F() local
135 Strategys inputs = {{4, 8}}; in TEST_F() local
146 Strategys inputs = {{4, 8}}; in TEST_F() local
158 Strategys inputs = {{1, 4, 8}}; in TEST_F() local
166 Strategys inputs = {{2, 4, 8}, {2, 4, 8}}; in TEST_F() local
174 Strategys inputs = {{4, 8}}; in TEST_F() local
|
| D | activation_info_test.cc | 67 Strategys inputs = {{2, 4, 8, 16}}; in TEST_F() local
82 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F() local
103 std::vector<TensorInfo> inputs = activation->inputs_tensor_info(); in TEST_F() local
120 Strategys inputs = {{2, 4, 8, 16}}; in TEST_F() local
131 Strategys inputs = {{1, 4, 8, 16}}; in TEST_F() local
151 Strategys inputs = {{2, 4, 8, 16}}; in TEST_F() local
164 Strategys inputs = {{2, 2, 8, 16}, {2, 4, 16, 1}}; in TEST_F() local
173 Strategys inputs = {{2, 4, 8}}; in TEST_F() local
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 57 std::vector<AnfNodePtr> inputs{value_node}; in TEST_F() local
102 std::vector<AnfNodePtr> inputs{NewValueNode(add_primitive)}; in TEST_F() local
112 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F() local
124 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F() local
133 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F() local
222 std::vector<AnfNodePtr> inputs; in TEST_F() local
246 std::vector<AnfNodePtr> inputs = {NewValueNode(prim::kPrimAdd), kernel_graph->NewParameter(), in TEST_F() local
267 std::vector<AnfNodePtr> inputs = {NewValueNode(prim::kPrimAdd), kernel_graph->NewParameter(), in TEST_F() local
297 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd), pre_add}; in TEST_F() local
325 std::vector<AnfNodePtr> inputs; in TEST_F() local
[all …]
|
| /third_party/mindspore/tests/ut/python/nn/ |
| D | test_cell.py | 30 def construct(self, *inputs):
39 def construct(self, *inputs):
49 def construct(self, *inputs):
64 def construct(self, *inputs):
75 def construct(self, *inputs):
167 def construct(self, *inputs):
180 def construct(self, *inputs):
193 def construct(self, *inputs):
260 def construct(self, *inputs):
294 def construct(self, *inputs):
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | depthwise_conv2d_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
74 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
120 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
167 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
214 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
261 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
308 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
355 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
402 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
449 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
[all …]
|
| D | conv2d_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
73 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
118 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
164 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
210 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
256 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
302 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
348 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
394 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
440 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
[all …]
|
| D | reshape_infer_test.cc | 29 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
55 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
87 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
119 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
151 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
183 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
215 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
249 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
283 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
318 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
|
| D | strided_slice_infer_test.cc | 28 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
70 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
112 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
154 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
186 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
219 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
272 std::vector<TensorC *> inputs(inputs_size, NULL); in TEST_F() local
|
| /third_party/mindspore/tests/ut/python/pynative_mode/nn/ |
| D | test_cell.py | 31 def construct(self, *inputs):
42 def construct(self, *inputs):
54 def construct(self, *inputs):
69 def construct(self, *inputs):
82 def construct(self, *inputs):
178 def construct(self, *inputs):
193 def construct(self, *inputs):
208 def construct(self, *inputs):
272 def construct(self, *inputs):
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensor_array.h | 29 TensorArrayCPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in TensorArrayCPUKernel()
48 TensorArrayBaseCPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in TensorArrayBaseCPUKernel()
64 TensorArrayReadCPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in TensorArrayReadCPUKernel()
76 TensorArrayWriteCPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in TensorArrayWriteCPUKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 102 const auto &inputs = node->inputs(); in ConvertGetAttrToTupleGetItem() local
148 const auto &inputs = node->inputs(); in ConvertDictGetItemToTupleGetItem() local
188 const auto &inputs = node->inputs(); in ConvertDictSetItemToTupleSetItem() local
234 std::vector<AnfNodePtr> inputs; in ConvertMakeRecordToMakeTuple() local
245 const auto &inputs = node->inputs(); in ErasePartialNode() local
276 std::vector<AnfNodePtr> inputs; in ConvertMakeListToMakeTuple() local
287 const auto &inputs = node->inputs(); in ConvertListGetItemToTupleGetItem() local
306 const auto &inputs = node->inputs(); in ConvertListSetItemToTupleSetItem() local
323 const auto &inputs = node->inputs(); in EraseMakeDictNode() local
331 const auto &inputs = node->inputs(); in EraseDictGetValues() local
[all …]
|