Home
last modified time | relevance | path

Searched defs:weights_tensor (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/string/
Dpredict.cc39 auto weights_tensor = in_tensors_[WEIGHT_INDEX]; in Prepare() local
77 auto weights_tensor = in_tensors_[WEIGHT_INDEX]; in GetLabelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/
Dlstm_grad_data_fp32.cc58 auto weights_tensor = in_tensors_.at(weights_index); in Run() local
Dlstm_grad_fp32.cc60 auto weights_tensor = in_tensors_.at(weights_index); in Run() local
/third_party/mindspore/test/unittest/
Dmindspore_unit_test.cpp657 OH_AI_TensorHandle weights_tensor = get_update_weights.handle_list[i]; variable
696 OH_AI_TensorHandle weights_tensor = export_update_weights.handle_list[i]; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/api/
Dmodel_c_test.cc401 MSTensorHandle weights_tensor = get_update_weights.handle_list[i]; in TEST() local