Searched defs:logsoftmax (Results 1 – 2 of 2) sorted by relevance
3109 Output logsoftmax = ops::Log(s.WithOpName("output"), softmax); in TEST_F() local
4938 auto logsoftmax = ops::LogSoftmax(s.WithOpName("my_logsoftmax"), input); in TEST_P() local