Searched refs:softmax (Results 1 – 3 of 3) sorted by relevance
16 let result = softmax(output_buffer); in main()23 fn softmax(output_tensor: Vec<f32>) -> Vec<f32> { in softmax() function
32 let result = softmax(output_buffer); in main()74 fn softmax(output_tensor: Vec<f32>) -> Vec<f32> { in softmax() function