Home
last modified time | relevance | path

Searched refs:compute_transposed_shape (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClTransposeKernel.cpp56 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in configure()
93 … dst_info = src->clone()->set_tensor_shape(misc::shape_calculator::compute_transposed_shape(*src)); in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNELSTMLayer.cpp223 TensorShape cell_state1_shape = compute_transposed_shape(*recurrent_to_output_weights->info()); in configure()
455 …TensorShape units_out_transposed_shape = compute_transposed_shape(*recurrent_to_output_weight… in validate()
456 TensorShape num_units_transposed_shape = compute_transposed_shape(*forget_gate_bias); in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLLSTMLayer.cpp244 TensorShape cell_state1_shape = compute_transposed_shape(*recurrent_to_output_weights->info()); in configure()
480 …TensorShape units_out_transposed_shape = compute_transposed_shape(*recurrent_to_output_weight… in validate()
481 TensorShape num_units_transposed_shape = compute_transposed_shape(*forget_gate_bias); in validate()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuTransposeKernel.cpp432 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in configure()
468 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in validate()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClFullyConnected.cpp343 …lone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_transposed_shape(*weights)… in validate()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuFullyConnected.cpp384 …lone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_transposed_shape(*weights)… in validate()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h403 inline TensorShape compute_transposed_shape(const ITensorInfo &input) in compute_transposed_shape() function