Home
last modified time | relevance | path

Searched refs:allTensorSize (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dsplit_builder.cpp38 auto allTensorSize = allTensors.size(); in SetInputAndOutput() local
39 …bool isOverTensorSize = std::any_of(inputsIndex.begin(), inputsIndex.end(), [allTensorSize](uint32… in SetInputAndOutput()
40 return index >= allTensorSize; in SetInputAndOutput()
47 …isOverTensorSize = std::any_of(outputsIndex.begin(), outputsIndex.end(), [allTensorSize](uint32_t … in SetInputAndOutput()
48 return index >= allTensorSize; in SetInputAndOutput()
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dneural_network_runtime.cpp327 size_t allTensorSize = liteGraph->all_tensors_.size(); // LiteGraph所有tensor个数 in ParseInputDimsFromExtensions() local
332 if (liteGraph->input_indices_[i] >= allTensorSize) { in ParseInputDimsFromExtensions()