Home
last modified time | relevance | path

Searched refs:input_indexes (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/lite/src/train/
Dclassification_train_accuracy_monitor.cc28 … const std::vector<int> &input_indexes, in ClassificationTrainAccuracyMonitor() argument
30 …accuracy_metrics_ = std::make_shared<AccuracyMetrics>(accuracy_metrics, input_indexes, output_inde… in ClassificationTrainAccuracyMonitor()
60 auto input_indexes = accuracy_metrics_->input_indexes_; in StepEnd() local
62 for (unsigned int i = 0; i < input_indexes.size(); i++) { in StepEnd()
63 if ((inputs.size() <= static_cast<unsigned int>(input_indexes[i])) || in StepEnd()
65 MS_LOG(WARNING) << "indices " << input_indexes[i] << "/" << output_indexes[i] in StepEnd()
69 if (inputs.at(input_indexes[i])->data_type() == kNumberTypeInt32) { in StepEnd()
70 …accuracy += CalculateSparseClassification(inputs.at(input_indexes[i]), outputs.at(output_indexes[i… in StepEnd()
72 …accuracy += CalculateOneHotClassification(inputs.at(input_indexes[i]), outputs.at(output_indexes[i… in StepEnd()
Daccuracy_metrics.cc25 AccuracyMetrics::AccuracyMetrics(int accuracy_metrics, const std::vector<int> &input_indexes, in AccuracyMetrics() argument
28 if (input_indexes.size() == output_indexes.size()) { in AccuracyMetrics()
29 input_indexes_ = input_indexes; in AccuracyMetrics()
Dtrain_export.cc43 auto input_indexes = old_node->inputIndex; in GetLinkedPostIdx() local
44 if (IsContain<uint32_t>(input_indexes, tensor_idx)) { in GetLinkedPostIdx()
/third_party/ffmpeg/tests/dnn/
Ddnn-layer-mathbinary-test.c55 int32_t input_indexes[1]; in test_broadcast_input0() local
73 input_indexes[0] = 0; in test_broadcast_input0()
74 ff_dnn_execute_layer_math_binary(operands, input_indexes, 1, &params, NULL); in test_broadcast_input0()
95 int32_t input_indexes[1]; in test_broadcast_input1() local
113 input_indexes[0] = 0; in test_broadcast_input1()
114 ff_dnn_execute_layer_math_binary(operands, input_indexes, 1, &params, NULL); in test_broadcast_input1()
135 int32_t input_indexes[2]; in test_no_broadcast() local
160 input_indexes[0] = 0; in test_no_broadcast()
161 input_indexes[1] = 1; in test_no_broadcast()
162 ff_dnn_execute_layer_math_binary(operands, input_indexes, 2, &params, NULL); in test_no_broadcast()
Ddnn-layer-pad-test.c48 int32_t input_indexes[1]; in test_with_mode_symmetric() local
81 input_indexes[0] = 0; in test_with_mode_symmetric()
82 ff_dnn_execute_layer_pad(operands, input_indexes, 1, &params, NULL); in test_with_mode_symmetric()
118 int32_t input_indexes[1]; in test_with_mode_reflect() local
146 input_indexes[0] = 0; in test_with_mode_reflect()
147 ff_dnn_execute_layer_pad(operands, input_indexes, 1, &params, NULL); in test_with_mode_reflect()
183 int32_t input_indexes[1]; in test_with_mode_constant() local
212 input_indexes[0] = 0; in test_with_mode_constant()
213 ff_dnn_execute_layer_pad(operands, input_indexes, 1, &params, NULL); in test_with_mode_constant()
Ddnn-layer-avgpool-test.c53 int32_t input_indexes[1]; in test_with_same() local
93 input_indexes[0] = 0; in test_with_same()
94 ff_dnn_execute_layer_avg_pool(operands, input_indexes, 1, &params, NULL); in test_with_same()
136 int32_t input_indexes[1]; in test_with_valid() local
173 input_indexes[0] = 0; in test_with_valid()
174 ff_dnn_execute_layer_avg_pool(operands, input_indexes, 1, &params, NULL); in test_with_valid()
Ddnn-layer-conv2d-test.c64 int32_t input_indexes[1]; in test_with_same_dilate() local
120 input_indexes[0] = 0; in test_with_same_dilate()
121 ff_dnn_execute_layer_conv2d(operands, input_indexes, 1, &params, &ctx); in test_with_same_dilate()
172 int32_t input_indexes[1]; in test_with_valid() local
224 input_indexes[0] = 0; in test_with_valid()
225 ff_dnn_execute_layer_conv2d(operands, input_indexes, 1, &params, &ctx); in test_with_valid()
Ddnn-layer-depth2space-test.c53 int32_t input_indexes[1]; in test() local
82 input_indexes[0] = 0; in test()
84 ff_dnn_execute_layer_depth2space(operands, input_indexes, 1, &params, NULL); in test()
Ddnn-layer-maximum-test.c32 int32_t input_indexes[1]; in test() local
47 input_indexes[0] = 0; in test()
48 ff_dnn_execute_layer_maximum(operands, input_indexes, 1, &params, NULL); in test()
Ddnn-layer-dense-test.c64 int32_t input_indexes[1]; in test() local
109 input_indexes[0] = 0; in test()
110 ff_dnn_execute_layer_dense(operands, input_indexes, 1, &params, NULL); in test()
Ddnn-layer-mathunary-test.c75 int32_t input_indexes[1]; in test() local
89 input_indexes[0] = 0; in test()
90 ff_dnn_execute_layer_math_unary(operands, input_indexes, 1, &params, NULL); in test()
/third_party/mindspore/mindspore/ccsrc/utils/
Dconfig_manager.h41 … const std::vector<std::vector<int64_t>> &shapes, const std::vector<int64_t> &input_indexes) in DatasetGraphParam() argument
47 input_indexes_(input_indexes) {} in DatasetGraphParam()
62 std::vector<int64_t> input_indexes() const { return input_indexes_; } in input_indexes() function
/third_party/mindspore/mindspore/lite/src/cxx_api/metrics/
Daccuracy.cc26 AccuracyMetrics::AccuracyMetrics(int accuracy_metrics, const std::vector<int> &input_indexes, in AccuracyMetrics() argument
29 …MetricsImpl(new (std::nothrow) lite::AccuracyMetrics(accuracy_metrics, input_indexes, output_index… in AccuracyMetrics()
/third_party/mindspore/mindspore/lite/src/cxx_api/callback/
Dtrain_accuracy.cc26 …racy::TrainAccuracy(int print_every_n, int accuracy_metrics, const std::vector<int> &input_indexes, in TrainAccuracy() argument
29 print_every_n, accuracy_metrics, input_indexes, output_indexes)); in TrainAccuracy()
/third_party/mindspore/tests/st/pynative/loss_scale/
Dtest_loss_scale.py38 np_types=None, output_shapes=None, input_indexes=(), func_name=''): argument
44 self._input_indexes = input_indexes
67 def input_indexes(self): member in MindData
105 input_indexes=(0, 1), func_name='')
/third_party/mindspore/include/api/metrics/
Daccuracy.h28 …Metrics(int accuracy_metrics = METRICS_CLASSIFICATION, const std::vector<int> &input_indexes = {1},
/third_party/mindspore/include/api/callback/
Dtrain_accuracy.h35 const std::vector<int> &input_indexes = {1},
/third_party/mindspore/mindspore/lite/include/train/
Daccuracy_metrics.h31 …Metrics(int accuracy_metrics = METRICS_CLASSIFICATION, const std::vector<int> &input_indexes = {1},
Dclassification_train_accuracy_monitor.h36 const std::vector<int> &input_indexes = {1},
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline_ge.h49 const std::vector<int64_t> &input_indexes, const std::string &phase);
Dpipeline.h168 … const std::vector<int64_t> &input_indexes, const std::string &phase, bool need_run);
173 const std::vector<int64_t> &input_indexes, bool need_run);
Dpipeline.cc1149 … const std::vector<int64_t> &input_indexes, const std::string &phase, bool need_run) { in InitExecDataset() argument
1162 …return InitExecDatasetVm(queue_name, iter_num, batch_size, types, shapes, input_indexes, need_run); in InitExecDataset()
1165 return InitExecDatasetGe(queue_name, iter_num, batch_size, types, shapes, input_indexes, phase); in InitExecDataset()
1177 const std::vector<int64_t> &input_indexes, bool need_run) { in InitExecDatasetVm() argument
1186 …(void)std::transform(input_indexes.begin(), input_indexes.end(), std::back_inserter(int_input_inde… in InitExecDatasetVm()
Dpipeline_ge.cc114 const std::vector<int64_t> &input_indexes, const std::string &phase) { in InitExecDatasetGe() argument
124 DatasetGraphParam param(queue_name, size, batch_size, ge_types, shapes, input_indexes); in InitExecDatasetGe()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc293 if (!param.input_indexes().empty() && input_idx <= param.input_indexes().size()) { in MakeDatasetHandler()
294 getnext_idx = param.input_indexes()[input_idx] - 1; // input_idx start from 0. in MakeDatasetHandler()