Home
last modified time | relevance | path

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

/external/armnn/src/backends/neon/workloads/
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp297 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload() local
300 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()
314 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload() local
315 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()
736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
739 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
753 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
754 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp283 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload() local
286 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()
300 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload() local
301 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()
718 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
721 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
735 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
736 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
/external/armnn/src/backends/cl/workloads/
DClUnidirectionalSequenceLstmFloatWorkload.cpp296 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload() local
298 (&input)->info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()
311 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload() local
312 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()
736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
738 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
751 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
752 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()