• Home
  • Raw
  • Download

Lines Matching refs:tensorIndex

188                  size_t tensorIndex,  in CheckTensor()  argument
200 if (tensorIndex >= model->subgraphs[subgraphIndex]->tensors.size()) in CheckTensor()
207 tensorIndex, in CheckTensor()
620 armnn::LayerBindingId GenerateLayerBindingId(size_t subgraphIndex, size_t tensorIndex) in GenerateLayerBindingId() argument
624 return static_cast<armnn::LayerBindingId>((tensorIndex<<8)+subgraphIndex); in GenerateLayerBindingId()
1043 …for (size_t tensorIndex = 0; tensorIndex < m_SubgraphConnections[subgraphIndex].size(); ++tensorIn… in CreateNetworkFromModel() local
1045 if (m_SubgraphConnections[subgraphIndex][tensorIndex].outputSlot != nullptr) in CreateNetworkFromModel()
1048 … inputSlotIdx < m_SubgraphConnections[subgraphIndex][tensorIndex].inputSlots.size(); in CreateNetworkFromModel()
1051 m_SubgraphConnections[subgraphIndex][tensorIndex].outputSlot->Connect( in CreateNetworkFromModel()
1052 … *(m_SubgraphConnections[subgraphIndex][tensorIndex].inputSlots[inputSlotIdx])); in CreateNetworkFromModel()
1073 size_t tensorIndex, in RegisterProducerOfTensor() argument
1076 CHECK_TENSOR(m_Model, subgraphIndex, tensorIndex); in RegisterProducerOfTensor()
1078 ARMNN_ASSERT(m_SubgraphConnections[subgraphIndex].size() > tensorIndex); in RegisterProducerOfTensor()
1080 TensorSlots & tensorSlots = m_SubgraphConnections[subgraphIndex][tensorIndex]; in RegisterProducerOfTensor()
1091 tensorIndex, in RegisterProducerOfTensor()
1099 size_t tensorIndex, in RegisterConsumerOfTensor() argument
1102 CHECK_TENSOR(m_Model, subgraphIndex, tensorIndex); in RegisterConsumerOfTensor()
1104 ARMNN_ASSERT(m_SubgraphConnections[subgraphIndex].size() > tensorIndex); in RegisterConsumerOfTensor()
1106 TensorSlots& tensorSlots = m_SubgraphConnections[subgraphIndex][tensorIndex]; in RegisterConsumerOfTensor()
3800 unsigned int tensorIndex = outputTensorIndexes[0]; in ParseUnidirectionalSequenceLSTM() local
3802 RegisterProducerOfTensor(subgraphIndex, tensorIndex, slot); in ParseUnidirectionalSequenceLSTM()
4909 unsigned int tensorIndex = tensorIndexes[index]; in RegisterInputSlots() local
4911 RegisterConsumerOfTensor(subgraphIndex, tensorIndex, slot); in RegisterInputSlots()
4936 unsigned int tensorIndex = tensorIndexes[slotIndex]; in RegisterOutputSlots() local
4938 RegisterProducerOfTensor(subgraphIndex, tensorIndex, slot); in RegisterOutputSlots()
5000 …for (unsigned int tensorIndex = 0; tensorIndex < m_SubgraphConnections[subgraphIndex].size(); ++te… in SetupConstantLayerTensorInfos() local
5002 if (m_SubgraphConnections[subgraphIndex][tensorIndex].outputSlot == nullptr && in SetupConstantLayerTensorInfos()
5003 m_SubgraphConnections[subgraphIndex][tensorIndex].inputSlots.size() > 0) in SetupConstantLayerTensorInfos()
5005 TensorRawPtr tensorPtr = subgraphPtr->tensors[tensorIndex].get(); in SetupConstantLayerTensorInfos()
5009 m_TensorInfos.insert({tensorIndex, tensorInfo}); in SetupConstantLayerTensorInfos()
5022 …for (unsigned int tensorIndex = 0; tensorIndex < m_SubgraphConnections[subgraphIndex].size(); ++te… in SetupConstantLayers() local
5024 if (m_SubgraphConnections[subgraphIndex][tensorIndex].outputSlot == nullptr && in SetupConstantLayers()
5025 m_SubgraphConnections[subgraphIndex][tensorIndex].inputSlots.size() > 0) in SetupConstantLayers()
5027 TensorRawPtr tensorPtr = subgraphPtr->tensors[tensorIndex].get(); in SetupConstantLayers()
5048 { tensorIndex }); in SetupConstantLayers()
5050 else if (ShouldConstantTensorBeCreated(tensorIndex)) in SetupConstantLayers()
5073 {tensorIndex}); in SetupConstantLayers()
5111 bool TfLiteParserImpl::ShouldConstantTensorBeCreated(unsigned int tensorIndex) in ShouldConstantTensorBeCreated() argument
5114 return (std::find(m_ConstantsToBeCreated.begin(), m_ConstantsToBeCreated.end(), tensorIndex) in ShouldConstantTensorBeCreated()