| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEGenerateProposalsLayer.cpp | 239 …TensorInfo all_anchors_info(anchors->clone()->set_tensor_shape(TensorShape(values_per_roi, total_n… in validate()
242 …TensorInfo deltas_permuted_info = deltas->clone()->set_tensor_shape(TensorShape(values_per_roi * n… in validate()
243 …TensorInfo scores_permuted_info = scores->clone()->set_tensor_shape(TensorShape(num_anchors, feat_… in validate()
255 …TensorInfo deltas_flattened_info(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi, tot… in validate()
258 …TensorInfo scores_flattened_info(scores->clone()->set_tensor_shape(TensorShape(1, total_num_anchor… in validate()
259 …TensorInfo proposals_4_roi_values(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi, to… in validate()
264 …TensorInfo proposals_4_roi_values_quantized(deltas->clone()->set_tensor_shape(TensorShape(values_… in validate()
268 …TensorInfo all_anchors_f32_info(anchors->clone()->set_tensor_shape(TensorShape(values_per_roi, tot… in validate()
271 …TensorInfo deltas_flattened_f32_info(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi,… in validate()
274 …TensorInfo proposals_4_roi_values_f32(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi… in validate()
|
| D | NEReductionOperation.cpp | 79 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate()
90 …info_before_reshape.set_data_type(output_data_type).set_tensor_shape(shape_before_reshape).set_num… in validate()
123 …tor()->init(input->info()->clone()->set_data_type(output_data_type).set_tensor_shape(output_intern… in configure()
127 …ut->info(), input->info()->clone()->set_data_type(output_data_type).set_tensor_shape(output_extern… in configure()
|
| D | NEFlattenLayer.cpp | 54 …auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(misc::shape_calculato… in configure()
65 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(misc::shape_calculator::com… in validate()
|
| D | NEL2NormalizeLayer.cpp | 69 sum_sq.set_tensor_shape(shape); in validate()
76 sum_sq.set_tensor_shape(shape); in validate()
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuGemmMatrixMultiplyKernel.cpp | 97 const TensorInfo tensor_info0 = lhs->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
98 …const TensorInfo tensor_info_reshaped0 = lhs->clone()->set_tensor_shape(misc::shape_calculator::co… in validate_arguments()
107 const TensorInfo tensor_info1 = rhs->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
108 …const TensorInfo tensor_info_reshaped1 = rhs->clone()->set_tensor_shape(misc::shape_calculator::co… in validate_arguments()
137 auto_init_if_empty(*dst, lhs->clone()->set_tensor_shape(tensor_shape)); in configure()
|
| /external/ComputeLibrary/src/core/helpers/ |
| D | AutoConfiguration.h | 51 info.set_tensor_shape(shape);
72 info_sink.set_tensor_shape(info_source.tensor_shape()); in auto_init_if_empty()
92 info.set_tensor_shape(shape); in set_shape_if_empty()
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLGenerateProposalsLayer.cpp | 237 …TensorInfo all_anchors_info(anchors->clone()->set_tensor_shape(TensorShape(values_per_roi, total_n… in validate()
240 …TensorInfo deltas_permuted_info = deltas->clone()->set_tensor_shape(TensorShape(values_per_roi * n… in validate()
241 …TensorInfo scores_permuted_info = scores->clone()->set_tensor_shape(TensorShape(num_anchors, feat_… in validate()
253 …TensorInfo deltas_flattened_info(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi, tot… in validate()
256 …TensorInfo scores_flattened_info(scores->clone()->set_tensor_shape(TensorShape(1, total_num_anchor… in validate()
257 …TensorInfo proposals_4_roi_values(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi, to… in validate()
262 …TensorInfo proposals_4_roi_values_quantized(deltas->clone()->set_tensor_shape(TensorShape(values_… in validate()
266 …TensorInfo all_anchors_f32_info(anchors->clone()->set_tensor_shape(TensorShape(values_per_roi, tot… in validate()
269 …TensorInfo deltas_flattened_f32_info(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi,… in validate()
272 …TensorInfo proposals_4_roi_values_f32(deltas->clone()->set_tensor_shape(TensorShape(values_per_roi… in validate()
|
| D | CLReductionOperation.cpp | 58 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate()
72 …ti.set_data_type(data_type).set_tensor_shape(shape).set_num_channels(num_channels).set_quantizatio… in validate()
102 _unreshaped_output.allocator()->init(input->info()->clone()->set_tensor_shape(shape)); in configure_intermediate_result_vector()
124 …auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(output_shape).set_dat… in configure()
|
| D | CLArgMinMaxLayer.cpp | 66 …const TensorInfo expected_output_shape = output->clone()->set_tensor_shape(arm_compute::misc::shap… in validate()
74 …ti.set_data_type(data_type).set_tensor_shape(shape).set_num_channels(num_channels).set_quantizatio… in validate()
95 sums_vector[i].set_tensor_shape(shape); in validate()
131 …auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(output_shape).set_dat… in configure()
155 …_results_vector[i].allocator()->init(input->info()->clone()->set_tensor_shape(shape).set_data_type… in configure()
|
| D | CLGEMMDeconvolutionLayer.cpp | 151 …input_info = input->clone()->set_is_resizable(true).reset_padding().set_tensor_shape(nhwc_input_sh… in validate()
153 …hts_info = weights->clone()->set_is_resizable(true).reset_padding().set_tensor_shape(nhwc_weights_… in validate()
160 …const TensorInfo reshaped_info = weights->clone()->set_tensor_shape(reshaped_shape).set_data_lay… in validate()
164 …nfo reshaped_t_info = reshaped_info.clone()->set_is_resizable(true).set_tensor_shape(transposed_sh… in validate()
172 …TensorInfo gemm_output_info = reshaped_t_info.clone()->set_tensor_shape(gemm_output_shape).set_is_… in validate()
179 …RN_ON_ERROR(CLGEMMLowpMatrixMultiplyCore::validate(&input->clone()->set_tensor_shape(nhwc_input_sh… in validate()
185 …ARM_COMPUTE_RETURN_ON_ERROR(CLGEMM::validate(&input->clone()->set_tensor_shape(nhwc_input_shape).s… in validate()
191 …TensorInfo col2im_output_info = gemm_output_info.clone()->set_tensor_shape(deconv_shape).… in validate()
|
| D | CLReduceMean.cpp | 88 const TensorInfo out_info = input->clone()->set_tensor_shape(out_shape); in validate_config()
122 auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(output_shape)); in configure()
136 TensorInfo output_no_quant_info = input->info()->clone()->set_tensor_shape(output_shape); in configure()
187 … auto_init_if_empty(*tmp_output->info(), tmp_input->info()->clone()->set_tensor_shape(out_shape)); in configure()
|
| D | CLFlattenLayer.cpp | 62 …auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(misc::shape_calculato… in configure()
73 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(misc::shape_calculator::com… in validate()
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClSoftmax.cpp | 77 _max_info = tmp_input_info.clone()->set_tensor_shape(max_sum_shape); in configure()
78 …_sum_info = tmp_input_info.clone()->set_tensor_shape(max_sum_shape).set_data_type(… in configure()
112 TensorInfo input_permuted(src.clone()->set_tensor_shape(permuted_shape)); in validate()
114 TensorInfo output_permuted(dst.clone()->set_tensor_shape(permuted_shape)); in validate()
124 TensorInfo tensor_info_max(src.clone()->set_tensor_shape(max_sum_shape).set_is_resizable(true)); in validate()
125 …TensorInfo tensor_info_sum(src.clone()->set_tensor_shape(max_sum_shape).set_data_type(tmp_data_typ… in validate()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClGemmLowpMatrixMultiplyReshapedKernel.cpp | 83 const TensorInfo tensor_info0 = src0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
84 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
86 …const TensorInfo tensor_info_reshaped0 = src0->clone()->set_tensor_shape(compute_lhs_reshaped_shap… in validate_arguments()
87 …const TensorInfo tensor_info_reshaped1 = src1->clone()->set_tensor_shape(compute_rhs_reshaped_shap… in validate_arguments()
94 …const TensorInfo tensor_info_dst = dst->clone()->set_tensor_shape(compute_mm_shape(*src0, *src1, g… in validate_arguments()
111 …auto_init_if_empty(*dst, src0->clone()->set_tensor_shape(compute_mm_shape(*src0, *src1, gemm_info)… in validate_and_configure_window()
120 tmp_info.set_tensor_shape(tmp_shape); in validate_and_configure_window()
|
| D | ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 131 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
132 …const TensorInfo tensor_info_reshaped1 = src1->clone()->set_tensor_shape(misc::shape_calculator::c… in validate_arguments()
138 …const TensorInfo tensor_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_… in validate_arguments()
154 …auto_init_if_empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(… in validate_and_configure_window()
164 tmp_info.set_tensor_shape(tmp_shape); in validate_and_configure_window()
214 …auto_init_if_empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(… in configure()
|
| D | ClGemmMatrixMultiplyReshapedKernel.cpp | 127 const TensorInfo tensor_info0 = src0->clone()->set_tensor_shape(tensor_shape0); in validate_arguments()
128 const TensorInfo tensor_info1 = src1->clone()->set_tensor_shape(tensor_shape1); in validate_arguments()
130 …const TensorInfo tensor_info_reshaped0 = src0->clone()->set_tensor_shape(misc::shape_calculator::c… in validate_arguments()
131 …const TensorInfo tensor_info_reshaped1 = src1->clone()->set_tensor_shape(misc::shape_calculator::c… in validate_arguments()
138 …const TensorInfo tensor_info_dst = dst->clone()->set_tensor_shape(misc::shape_calculator::compute_… in validate_arguments()
164 tmp_info.set_tensor_shape(tmp_shape); in validate_and_configure_window()
195 …auto_init_if_empty(*dst, src0->clone()->set_tensor_shape(misc::shape_calculator::compute_mm_shape(… in configure()
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| D | GpuSoftmax.cpp | 71 TensorInfo logits(src->clone()->set_tensor_shape(logits_sum_shape)); in is_supported_op()
75 TensorInfo sum(src->clone()->set_tensor_shape(logits_sum_shape)); in is_supported_op()
139 … = sketch.implementation().create_auxiliary_tensor(src->clone()->set_tensor_shape(logits_sum_sh… in create_op()
141 …sum = sketch.implementation().create_auxiliary_tensor(src->clone()->set_tensor_shape(logits_sum_sh… in create_op()
|
| D | GpuReshape.cpp | 55 auto_init_if_empty(dst_info_to_validate, src->clone()->set_tensor_shape(attributes.shape())); in is_supported_op_helper()
99 auto_init_if_empty(dst_info_to_validate, src->clone()->set_tensor_shape(attributes.shape())); in validate_op()
125 auto_init_if_empty(*dst, src->clone()->set_tensor_shape(attributes.shape())); in create_op()
|
| D | GpuPool2d.cpp | 76 auto_init_if_empty(dst_info_to_validate, src->clone()->set_tensor_shape(shape)); in validate_op()
111 auto_init_if_empty(dst_info_to_validate, src->clone()->set_tensor_shape(shape)); in is_supported_op()
150 auto_init_if_empty(*dst, src->clone()->set_tensor_shape(shape)); in create_op()
|
| /external/ComputeLibrary/src/core/ |
| D | SubTensorInfo.cpp | 84 clone_obj->set_tensor_shape(_tensor_shape); in clone()
89 ITensorInfo &SubTensorInfo::set_tensor_shape(const TensorShape &shape) in set_tensor_shape() function in arm_compute::SubTensorInfo
103 _parent->set_tensor_shape(parent_extended_shape); in set_tensor_shape()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 298 (&input)->info()->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()
300 (&output)->info()->set_tensor_shape(acl_output_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkload()
312 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); 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()
738 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
740 … const_cast<arm_compute::TensorInfo*>(&aclOutputInfo)->set_tensor_shape(acl_output_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
752 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
808 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 300 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()
303 output.info()->set_tensor_shape(acl_output_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkload()
315 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); 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()
739 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
742 … const_cast<arm_compute::TensorInfo*>(&aclOutputInfo)->set_tensor_shape(acl_output_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
754 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
811 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
[all …]
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 286 input.info()->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()
289 output.info()->set_tensor_shape(acl_output_shape_shrink); in NeonUnidirectionalSequenceLstmWorkload()
301 m_PermuteFirstOut.info()->set_tensor_shape(acl_input_shape_shrink); 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()
721 … const_cast<arm_compute::TensorInfo*>(&aclInputInfo)->set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
724 … const_cast<arm_compute::TensorInfo*>(&aclOutputInfo)->set_tensor_shape(acl_output_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
736 aclPermuteOutInfo.set_tensor_shape(acl_input_shape_shrink); in NeonUnidirectionalSequenceLstmWorkloadValidate()
784 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in NeonUnidirectionalSequenceLstmWorkloadValidate()
[all …]
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuSoftmax.cpp | 85 TensorInfo max_info(tmp_input->clone()->set_tensor_shape(max_sum_shape)); in configure()
134 …const TensorInfo tensor_info_max_sum(src->clone()->set_tensor_shape(max_sum_shape).set_data_type(t… in validate()
145 TensorInfo input_permuted(src->clone()->set_tensor_shape(permuted_shape)); in validate()
147 TensorInfo output_permuted(dst->clone()->set_tensor_shape(permuted_shape)); in validate()
|
| /external/ComputeLibrary/tests/validation/fixtures/UNIT/ |
| D | MemoryManagerFixture.h | 231 …src.allocator()->info().set_tensor_shape(TensorShape(128U, _cur_batches)).set_is_resizable(true).e… in compute_target()
233 …fc1.allocator()->info().set_tensor_shape(TensorShape(128U, _cur_batches)).set_is_resizable(true).e… in compute_target()
235 …dst.allocator()->info().set_tensor_shape(TensorShape(24U, _cur_batches)).set_is_resizable(true).ex… in compute_target()
367 src.allocator()->info().set_tensor_shape(TensorShape(1U, 1U, 112U, i)); in compute_target()
368 …fc.allocator()->info().set_tensor_shape(TensorShape(8U, i)).set_is_resizable(true).extend_padding(… in compute_target()
370 dst.allocator()->info().set_tensor_shape(TensorShape(8U, i)); in compute_target()
|