Searched refs:compute_gather_shape (Results 1 – 5 of 5) sorted by relevance
50 …TensorShape output_shape = arm_compute::misc::shape_calculator::compute_gather_shape(input->tensor… in validate_arguments()64 …TensorShape output_shape = arm_compute::misc::shape_calculator::compute_gather_shape(input->tensor… in validate_and_configure_window()
79 …TensorShape output_shape = arm_compute::misc::shape_calculator::compute_gather_shape(input->tensor… in validate_arguments()243 …const TensorShape output_shape = arm_compute::misc::shape_calculator::compute_gather_shape(input->… in configure()
43 …const TensorShape dst_shape = arm_compute::misc::shape_calculator::compute_gather_shape(src.shap… in gather()
87 …TensorShape output_shape = arm_compute::misc::shape_calculator::compute_gather_shape(input_sh… in compute_target()
1538 inline TensorShape compute_gather_shape(const TensorShape &input_shape, const TensorShape &indices_… in compute_gather_shape() function