Searched refs:BuildArmComputeTensorShape (Results 1 – 13 of 13) sorted by relevance
296 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload()297 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload()311 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkload()418 … m_ConcatInputs[i]->info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in ClUnidirectionalSequenceLstmFloatWorkload()453 (&output)->info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandBatchMajor)); in ClUnidirectionalSequenceLstmFloatWorkload()457 (&output)->info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in ClUnidirectionalSequenceLstmFloatWorkload()736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()737 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()751 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()807 auto acl_shape_expand = BuildArmComputeTensorShape(shapeExpandTimeMajor); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()[all …]
128 …indices.info()->set_tensor_shape(BuildArmComputeTensorShape({ keyIndices["W"], keyIndices["ND"] })… in ClGatherNdWorkload()180 input.info()->set_tensor_shape(BuildArmComputeTensorShape(paramsInfo.GetShape())); in ClGatherNdWorkload()
143 arm_compute::TensorShape inputXTensorInfo = armcomputetensorutils::BuildArmComputeTensorShape( in ClBatchMatMulWorkload()147 arm_compute::TensorShape inputYTensorInfo = armcomputetensorutils::BuildArmComputeTensorShape( in ClBatchMatMulWorkload()
297 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload()298 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload()314 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkload()421 … m_ConcatInputs[i]->info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in NeonUnidirectionalSequenceLstmFloatWorkload()455 output.info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandBatchMajor)); in NeonUnidirectionalSequenceLstmFloatWorkload()459 output.info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in NeonUnidirectionalSequenceLstmFloatWorkload()736 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()737 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()753 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()810 auto acl_shape_expand = BuildArmComputeTensorShape(shapeExpandTimeMajor); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()[all …]
283 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload()284 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload()300 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkload()402 … m_ConcatInputs[i]->info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in NeonUnidirectionalSequenceLstmWorkload()436 output.info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandBatchMajor)); in NeonUnidirectionalSequenceLstmWorkload()440 output.info()->set_tensor_shape(BuildArmComputeTensorShape(shapeExpandTimeMajor)); in NeonUnidirectionalSequenceLstmWorkload()718 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()719 auto acl_output_shape_shrink = BuildArmComputeTensorShape(outputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()735 auto acl_input_shape_shrink = BuildArmComputeTensorShape(inputShapeShrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()783 auto acl_shape_expand = BuildArmComputeTensorShape(shapeExpandTimeMajor); in NeonUnidirectionalSequenceLstmWorkloadValidate()[all …]
125 …indices.info()->set_tensor_shape(BuildArmComputeTensorShape({ keyIndices["W"], keyIndices["ND"] })… in NeonGatherNdWorkload()175 input.info()->set_tensor_shape(BuildArmComputeTensorShape(paramsInfo.GetShape())); in NeonGatherNdWorkload()
125 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape) in BuildArmComputeTensorShape() function176 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape, unsigned… in BuildArmComputeTensorShape() function203 … const arm_compute::TensorShape aclTensorShape = BuildArmComputeTensorShape(tensorInfo.GetShape()); in BuildArmComputeTensorInfo()225 …const arm_compute::TensorShape aclTensorShape = BuildArmComputeTensorShape(tensorInfo.GetShape(), … in BuildArmComputeTensorInfo()
37 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape);41 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape, unsigned…
26 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()36 …const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(par… in CreateSubTensorHandle()
24 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()34 …const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(par… in CreateSubTensorHandle()
227 …arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape); in CreateSubTensorHandle()237 …const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(par… in CreateSubTensorHandle()
25 …const arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensor… in CreateSubTensorHandle()36 …const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(par… in CreateSubTensorHandle()
90 …const arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensor… in CreateSubTensorHandle()101 …const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(par… in CreateSubTensorHandle()