Searched defs:SoftmaxOpModel (Results 1 – 3 of 3) sorted by relevance
23 class SoftmaxOpModel : public SingleOpModelWithHexagon { class25 SoftmaxOpModel(float softmax_beta, const TensorData& input) { in SoftmaxOpModel() function in tflite::SoftmaxOpModel53 TEST(SoftmaxOpModel, Softmax4DUint8) { in TEST() argument70 TEST(SoftmaxOpModel, Softmax4DUint8_MultipleBatch) { in TEST() argument91 TEST(SoftmaxOpModel, Softmax4DInt8) { in TEST() argument108 TEST(SoftmaxOpModel, Softmax4DInt8_MultipleBatch) { in TEST() argument
33 class SoftmaxOpModel : public SingleOpModel { class35 SoftmaxOpModel(int batches, int size, float beta) in SoftmaxOpModel() function in tflite::__anon676cc4200111::SoftmaxOpModel
1398 class SoftmaxOpModel : public SingleOpModelWithNNAPI { class1400 SoftmaxOpModel(const TensorData& input, float beta) { in SoftmaxOpModel() function in tflite::__anon8716bded0111::SoftmaxOpModel