Searched refs:calculate_concatenate_shape (Results 1 – 8 of 8) sorted by relevance
54 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(srcs_vect… in configure()140 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(srcs_vect… in validate()
113 … const TensorShape dst_shape = misc::shape_calculator::calculate_concatenate_shape(src_ptrs, axis); in compute_target()163 … const TensorShape dst_shape = misc::shape_calculator::calculate_concatenate_shape(src_ptrs, axis); in compute_reference()
54 … dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(src_vec… in configure()212 …TensorShape dst_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(src_vecto… in validate()
119 …ensorShape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(inputs_… in configure()129 …pe weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(weights… in configure()198 …tm_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(lstm_we… in configure()301 …ut_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(in_out_… in configure()496 …hape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(inputs_… in validate()528 …tm_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(lstm_we… in validate()579 …ut_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(in_out_… in validate()
81 …output_descriptor.shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(shapes,… in compute_output_descriptor()
129 TensorShape calculate_concatenate_shape(const std::vector<TensorShape> &input_shapes, size_t axis);
468 …hape concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(inputs_… in validate()502 …tm_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(lstm_we… in validate()553 …ut_weights_concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(in_out_… in validate()
1363 inline TensorShape calculate_concatenate_shape(const std::vector<T *> &input, size_t axis) in calculate_concatenate_shape() function