Searched refs:m_PermuteFirstOut (Results 1 – 6 of 6) sorted by relevance
| /external/armnn/src/backends/cl/workloads/ |
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in ClUnidirectionalSequenceLstmFloatWorkload()
75 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in ClUnidirectionalSequenceLstmFloatWorkload()
78 …layer->configure(clCompileContext, &input, &m_PermuteFirstOut, arm_compute::PermutationVector(0U,2… in ClUnidirectionalSequenceLstmFloatWorkload()
137 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in ClUnidirectionalSequenceLstmFloatWorkload()
309 TensorShape inputShape = GetTensorShape(m_PermuteFirstOut.info()->tensor_shape(), 1U); in ClUnidirectionalSequenceLstmFloatWorkload()
312 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()
313 inputLSTM = &m_PermuteFirstOut; in ClUnidirectionalSequenceLstmFloatWorkload()
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 76 arm_compute::CLTensor m_PermuteFirstOut; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 72 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmFloatWorkload()
73 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in NeonUnidirectionalSequenceLstmFloatWorkload()
76 layer->configure(&input, &m_PermuteFirstOut, arm_compute::PermutationVector(0U,2U,1U)); in NeonUnidirectionalSequenceLstmFloatWorkload()
136 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmFloatWorkload()
312 TensorShape inputShape = GetTensorShape(m_PermuteFirstOut.info()->tensor_shape(), 1U); in NeonUnidirectionalSequenceLstmFloatWorkload()
315 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()
316 inputLSTM = &m_PermuteFirstOut; in NeonUnidirectionalSequenceLstmFloatWorkload()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 74 BuildArmComputeTensor(m_PermuteFirstOut, permuteOutInfo); in NeonUnidirectionalSequenceLstmWorkload()
75 armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermuteFirstOut); in NeonUnidirectionalSequenceLstmWorkload()
78 layer->configure(&input, &m_PermuteFirstOut, arm_compute::PermutationVector(0U,2U,1U)); in NeonUnidirectionalSequenceLstmWorkload()
138 split_layer->configure(&m_PermuteFirstOut, m_SplitterOutputs, aclAxisSplit); in NeonUnidirectionalSequenceLstmWorkload()
298 TensorShape inputShape = GetTensorShape(m_PermuteFirstOut.info()->tensor_shape(), 1U); in NeonUnidirectionalSequenceLstmWorkload()
301 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()
302 inputLSTM = &m_PermuteFirstOut; in NeonUnidirectionalSequenceLstmWorkload()
|
| D | NeonUnidirectionalSequenceLstmWorkload.hpp | 70 arm_compute::Tensor m_PermuteFirstOut; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 74 arm_compute::Tensor m_PermuteFirstOut; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|