Home
last modified time | relevance | path

Searched refs:LogSoftmaxStage (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1810 class LogSoftmaxStage : public ArithmeticOptimizerStage { class
1812 explicit LogSoftmaxStage(const GraphOptimizerContext& ctx, in LogSoftmaxStage() function in tensorflow::grappler::__anonbb66c6470111::LogSoftmaxStage
1815 ~LogSoftmaxStage() override = default;
3585 pipeline.AddStage<LogSoftmaxStage>(ctx, ctx_ext); in SimplifyArithmeticOps()