Searched refs:acl_input_shape_shrink (Results 1 – 3 of 3) sorted by relevance
297 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload() local300 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()314 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload() local315 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local739 … 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() local754 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
283 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload() local286 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()300 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload() local301 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()718 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate() local721 … 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() local736 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
296 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload() local298 (&input)->info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()311 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload() local312 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local738 … 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() local752 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()