Searched refs:input_nums (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | apply_adagrad_d_a.cc | 46 const int64_t input_nums = 0; in InferShape() local
47 …CheckAndConvertUtils::CheckInteger("lr_shape size", SizeToInt(lr_shape.size()), kEqual, input_nums, in InferShape()
49 …CheckAndConvertUtils::CheckInteger("l1_shape size", SizeToInt(l1_shape.size()), kEqual, input_nums, in InferShape()
51 …CheckAndConvertUtils::CheckInteger("l2_shape size", SizeToInt(l2_shape.size()), kEqual, input_nums, in InferShape()
54 input_nums, primitive->name()); in InferShape()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | stack_base.cc | 62 auto input_nums = in_tensors_.size(); in ReSize() local
63 if (input_nums == 1) { in ReSize()
66 MS_ASSERT(input_nums > 1); in ReSize()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 324 def _concat_grad_uniform(input_shapes, input_nums): argument
327 for i in range(1, input_nums):
341 input_nums = len(x)
343 for i in range(input_nums):
345 is_uniform = _concat_grad_uniform(input_shapes, input_nums)
349 dx_tuple = P.Split(axis, input_nums)(dout)
353 for i in range(input_nums):
359 dx = P.Split(axis, input_nums)(dout)
361 for i in range(input_nums):
|
| D | grad_comm_ops.py | 515 input_nums = F.tuple_len(dout)
516 for i in range(input_nums):
522 input_nums = F.list_len(dout)
523 for i in range(input_nums):
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1946 size_t input_nums = AnfAlgo::GetInputTensorNum(node); in IsIndependentNode() local
1947 if (input_nums == 0) { in IsIndependentNode()
|