Searched refs:get_permutation_vector_from_softmax_axis (Results 1 – 4 of 4) sorted by relevance
69 … const auto perm_info = softmax_helpers::get_permutation_vector_from_softmax_axis(actual_axis); in configure()88 … const auto perm_info = softmax_helpers::get_permutation_vector_from_softmax_axis(actual_axis); in configure()110 …PermutationVector permutation_vector = softmax_helpers::get_permutation_vector_from_softmax_axis(a… in validate()
72 …input.configure(src, &_input_permuted, softmax_helpers::get_permutation_vector_from_softmax_axis(a… in configure()103 …tput.configure(&_output_permuted, dst, softmax_helpers::get_permutation_vector_from_softmax_axis(a… in configure()143 …PermutationVector permutation_vector = softmax_helpers::get_permutation_vector_from_softmax_axis(a… in validate()
46 PermutationVector get_permutation_vector_from_softmax_axis(size_t axis);
30 PermutationVector get_permutation_vector_from_softmax_axis(size_t axis) in get_permutation_vector_from_softmax_axis() function