Searched refs:concatInputsTensorInfosPtr (Results 1 – 3 of 3) sorted by relevance
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 562 std::vector<const arm_compute::ITensorInfo*> concatInputsTensorInfosPtr; in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
583 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
757 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
763 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
798 TensorShape shape = GetTensorShape(concatInputsTensorInfosPtr[0]->tensor_shape(), 1U); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
817 statusConcat = arm_compute::NEConcatenateLayer::validate(concatInputsTensorInfosPtr, in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
823 statusConcat = arm_compute::NEConcatenateLayer::validate(concatInputsTensorInfosPtr, in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
858 statusPermute2 = arm_compute::NEPermute::validate(concatInputsTensorInfosPtr[0], in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 543 std::vector<const arm_compute::ITensorInfo*> concatInputsTensorInfosPtr; in NeonUnidirectionalSequenceLstmWorkloadValidate() local
564 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in NeonUnidirectionalSequenceLstmWorkloadValidate()
739 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in NeonUnidirectionalSequenceLstmWorkloadValidate()
745 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in NeonUnidirectionalSequenceLstmWorkloadValidate()
771 TensorShape shape = GetTensorShape(concatInputsTensorInfosPtr[0]->tensor_shape(), 1U); in NeonUnidirectionalSequenceLstmWorkloadValidate()
790 statusConcat = arm_compute::NEConcatenateLayer::validate(concatInputsTensorInfosPtr, in NeonUnidirectionalSequenceLstmWorkloadValidate()
796 statusConcat = arm_compute::NEConcatenateLayer::validate(concatInputsTensorInfosPtr, in NeonUnidirectionalSequenceLstmWorkloadValidate()
831 statusPermute2 = arm_compute::NEPermute::validate(concatInputsTensorInfosPtr[0], in NeonUnidirectionalSequenceLstmWorkloadValidate()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 562 std::vector<const arm_compute::ITensorInfo*> concatInputsTensorInfosPtr; in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
583 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
754 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
760 outputLSTM = const_cast<arm_compute::ITensorInfo*>(concatInputsTensorInfosPtr[i]); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
795 TensorShape shape = GetTensorShape(concatInputsTensorInfosPtr[0]->tensor_shape(), 1U); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
814 statusConcat = arm_compute::CLConcatenateLayer::validate(concatInputsTensorInfosPtr, in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
820 statusConcat = arm_compute::CLConcatenateLayer::validate(concatInputsTensorInfosPtr, in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
854 statusPermute2 = arm_compute::CLPermute::validate(concatInputsTensorInfosPtr[0], in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
|