Searched defs:LogSoftmaxStage (Results 1 – 1 of 1) sorted by relevance
1903 class LogSoftmaxStage : public ArithmeticOptimizerStage { class1905 explicit LogSoftmaxStage(const GraphOptimizerContext& ctx, in LogSoftmaxStage() function in tensorflow::grappler::__anon327bfa1e0111::LogSoftmaxStage