Searched refs:compute_permutation_output_shape (Results 1 – 6 of 6) sorted by relevance
43 …const TensorShape output_shape = misc::shape_calculator::compute_permutation_output_shape(*input, … in validate_arguments()105 …const TensorShape output_shape = misc::shape_calculator::compute_permutation_output_shape(*input->… in configure()
108 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments()246 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in configure()
71 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments()
111 …sorShape permuted_shape = misc::shape_calculator::compute_permutation_output_shape(src, … in validate()
144 …sorShape permuted_shape = misc::shape_calculator::compute_permutation_output_shape(*src,… in validate()
109 inline TensorShape compute_permutation_output_shape(const ITensorInfo &input, const PermutationVect… in compute_permutation_output_shape() function