| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/Core/Src/ |
| D | main.c | 31 size_t inputs_num = inputs.size(); in main() local
32 void *inputs_binbuf[inputs_num]; in main()
33 int inputs_size[inputs_num]; in main()
34 for (size_t i = 0; i < inputs_num; ++i) { in main()
39 for (size_t i = 0; i < inputs_num; ++i) { in main()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/benchmark/ |
| D | load_input.c | 70 int ReadInputsFile(char *path, void **buffers, const int *inputs_size, int inputs_num) { in ReadInputsFile() argument
71 char *inputs_path[inputs_num]; in ReadInputsFile()
76 if (i >= inputs_num) { in ReadInputsFile()
77 printf("inputs num is error, need: %d\n", inputs_num); in ReadInputsFile()
85 for (i = 0; i < inputs_num; ++i) { in ReadInputsFile()
|
| D | benchmark.cc | 138 size_t inputs_num = inputs.size(); in main() local
139 void *inputs_binbuf[inputs_num]; in main()
140 int inputs_size[inputs_num]; in main()
141 for (size_t i = 0; i < inputs_num; ++i) { in main()
144 int ret = ReadInputsFile(const_cast<char *>(argv[1]), inputs_binbuf, inputs_size, inputs_num); in main()
148 for (size_t i = 0; i < inputs_num; ++i) { in main()
202 for (size_t i = 0; i < inputs_num; ++i) { in main()
|
| D | load_input.h | 29 int ReadInputsFile(char *path, void **buffers, const int *inputs_size, int inputs_num);
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/ |
| D | load_input.c | 70 int ReadInputsFile(char *path, void **buffers, const int *inputs_size, int inputs_num) { in ReadInputsFile() argument
71 char *inputs_path[inputs_num]; in ReadInputsFile()
76 if (i >= inputs_num) { in ReadInputsFile()
77 printf("inputs num is error, need: %d\n", inputs_num); in ReadInputsFile()
85 for (i = 0; i < inputs_num; ++i) { in ReadInputsFile()
|
| D | benchmark.cc | 138 size_t inputs_num = inputs.size(); in main() local
139 void *inputs_binbuf[inputs_num]; in main()
140 int inputs_size[inputs_num]; in main()
141 for (size_t i = 0; i < inputs_num; ++i) { in main()
144 int ret = ReadInputsFile(const_cast<char *>(argv[1]), inputs_binbuf, inputs_size, inputs_num); in main()
148 for (size_t i = 0; i < inputs_num; ++i) { in main()
202 for (size_t i = 0; i < inputs_num; ++i) { in main()
|
| D | load_input.h | 29 int ReadInputsFile(char *path, void **buffers, const int *inputs_size, int inputs_num);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_utils.cc | 27 std::vector<size_t> *inputs_num) { in GetInputsRealNum() argument
29 MS_EXCEPTION_IF_NULL(inputs_num); in GetInputsRealNum()
43 (*inputs_num).emplace_back(LongToSize(dyn_input_sizes[dyn_input_index])); in GetInputsRealNum()
47 (*inputs_num).emplace_back(1); in GetInputsRealNum()
|
| D | tbe_json_utils.h | 111 std::vector<size_t> *inputs_num);
|
| /third_party/mindspore/mindspore/lite/micro/coder/generator/component/ |
| D | train_component.cc | 126 size_t inputs_num = inputs.size(); in CodeTrainImplement() local
167 << " const void *input_ptr[" << inputs_num << "];\n" in CodeTrainImplement()
172 << "_SetInputs(input_ptr, " << inputs_num << ");\n" in CodeTrainImplement()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | stack_base.cc | 112 size_t inputs_num = in_tensors_.size(); in Run() local
113 all_inputs_ = static_cast<void **>(ms_context_->allocator->Malloc(inputs_num * sizeof(void *))); in Run()
118 for (size_t j = 0; j < inputs_num; ++j) { in Run()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kpynative.cc | 169 FuncGraphPtr BuildFakeBProp(const PrimitivePtr &prim, size_t inputs_num) { in BuildFakeBProp() argument
178 for (size_t i = 0; i < inputs_num; ++i) { in BuildFakeBProp()
897 auto inputs_num = cnode->size() - 1; in BuildBPropCutFuncGraph() local
912 for (size_t i = 0; i < inputs_num; ++i) { in BuildBPropCutFuncGraph()
927 auto inputs_num = cnode->size() - 1; in BuildMakeSequenceBprop() local
928 CacheKey key{prim->name(), inputs_num}; in BuildMakeSequenceBprop()
937 ss << "◀" << prim->ToString() << inputs_num; in BuildMakeSequenceBprop()
939 for (size_t i = 0; i < inputs_num; ++i) { in BuildMakeSequenceBprop()
960 for (size_t i = 0; i < inputs_num; ++i) { in BuildMakeSequenceBprop()
|
| D | kprim.cc | 664 auto inputs_num = cnode->first->cast<CNodePtr>()->size() - 1; in BpropCut() local
678 for (size_t i = 0; i < inputs_num; ++i) { in BpropCut()
703 auto inputs_num = cnode->first->cast<CNodePtr>()->inputs().size() - 1; in FakeBprop() local
713 if (inputs_num < monad_params_size) { in FakeBprop()
717 inputs_num -= monad_params_size; in FakeBprop()
726 for (size_t i = 0; i < inputs_num; ++i) { in FakeBprop()
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | block_util.py | 364 inputs_num = len(inputs)
365 if add_fake_input and inputs_num == 1:
367 inputs_num = 0
368 b = block_generator(block, inputs_num, desc_const=desc_const, const_first=const_first,
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | params_info.h | 50 size_t inputs_num() const;
|
| D | optimizer_kernel_factory.cc | 33 size_t input_num = params_info.inputs_num(); in Matched()
|
| D | aggregation_kernel_factory.cc | 33 size_t input_num = params_info.inputs_num(); in Matched()
|
| D | params_info.cc | 42 size_t ParamsInfo::inputs_num() const { return inputs_name_type_.size(); } in inputs_num() function in mindspore::fl::server::kernel::ParamsInfo
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | conv2d.cc | 45 int inputs_num = in_tensors_.size(); in CheckSpecs() local
46 if (inputs_num != INPUT_TENSOR_SIZE_2 && inputs_num != INPUT_TENSOR_SIZE_3) { in CheckSpecs()
47 MS_LOG(WARNING) << "Conv2D only supports 2 or 3 input Tensor but get " << inputs_num; in CheckSpecs()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | data_converter.cc | 135 …size_t inputs_num = py::cast<int64_t>(py::getattr(code_obj, "co_argcount")) - kBpropExcludeParamNu… in ConvertToBpropCut() local
136 for (size_t i = 0; i < inputs_num; ++i) { in ConvertToBpropCut()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 372 FuncGraphPtr BuildCondGraph(const AnfNodePtr &root_while_node, int inputs_num, const std::string &c… in BuildCondGraph() argument
377 for (int i = 0; i < inputs_num; i++) { in BuildCondGraph()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 2481 const size_t inputs_num = py::cast<int64_t>(py::getattr(code_obj, "co_argcount")) - 3; in DoGradForCustomBprop() local
2482 if (inputs_num > args.size()) { in DoGradForCustomBprop()
2483 …MS_EXCEPTION(TypeError) << "Size of bprop func inputs[" << inputs_num << "] is larger than size of… in DoGradForCustomBprop()
2488 for (size_t i = 0; i < inputs_num; i += 1) { in DoGradForCustomBprop()
|