Searched defs:softmax (Results 1 – 8 of 8) sorted by relevance
35 auto softmax = std::make_shared<Softmax>(); in TEST_F() local65 auto softmax = std::make_shared<Softmax>(); in TEST_F() local95 auto softmax = std::make_shared<Softmax>(); in TEST_F() local
86 auto softmax = VectorRef({is_softmax, q2k_normed_masked}); in DefineMPWithMaskPattern() local219 auto softmax = VectorRef({is_softmax, query_mul_key}); in DefineMPWithoutMaskPattern() local
239 def softmax(axis: int = -1) -> Callable: function
31 SoftmaxInfoPtr softmax; variable
536 def softmax(x, axis=None): function
1043 Softmax softmax{}; in ConvertSoftmax() local
381 static void wae5(const float *softmax, const float *el, in wae5()
1651 optional uint32 softmax = 4 [default = 0]; field