Searched refs:outputShapeShrink (Results 1 – 3 of 3) sorted by relevance
295 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkload() local298 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload()734 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local737 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
281 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in NeonUnidirectionalSequenceLstmWorkload() local284 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload()716 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in NeonUnidirectionalSequenceLstmWorkloadValidate() local719 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
295 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkload() local297 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload()735 TensorShape outputShapeShrink({outputShape[1], outputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local737 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()