Home
last modified time | relevance | path

Searched refs:kParameterWeightTensorMask (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h503 const int kParameterWeightTensorMask = 1; variable
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_session.cc1100 …_tensor_info->input_tensors_mask.emplace_back(parameter->has_default() ? kParameterWeightTensorMask in GetOpInputStubTensors()
1105 … input_tensor_info->input_tensors_mask.emplace_back(output_is_weight ? kParameterWeightTensorMask in GetOpInputStubTensors()
Dsession_basic.cc309 if (tensor_mask == kParameterWeightTensorMask) { in ConstructRunOpParameter()
1462 …ut_tensor_info->input_tensors_mask.emplace_back(tensor->is_parameter() ? kParameterWeightTensorMask in GetOpInputTensors()