Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc294 struct OptimizedNodesGroup { struct in tensorflow::grappler::__anonbb66c6470111::ArithmeticNodesGroupOptimizerStage
306 OptimizedNodesGroup group; in TrySimplify()
319 const OptimizedNodesGroup& group) = 0;
323 const OptimizedNodesGroup& group, const NodeDef& node) const = 0;
326 OptimizedNodesGroup* group) const { in AbsorbInputByOptimizedNodesGroup()
358 OptimizedNodesGroup* group) const { in CreateOptimizedNodesGroup()
402 void MarkAllMembersWithTag(const OptimizedNodesGroup& group, in MarkAllMembersWithTag()
410 bool IsOnTheSameDevice(const OptimizedNodesGroup& group, in IsOnTheSameDevice()
477 bool IsAbsorbableByOptimizedNodesGroup(const OptimizedNodesGroup& group, in IsAbsorbableByOptimizedNodesGroup()
514 string RewriteOptimizedNodesGroup(const OptimizedNodesGroup& group) override { in RewriteOptimizedNodesGroup()
[all …]