Home
last modified time | relevance | path

Searched defs:add_ops (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc446 const llvm::SmallVectorImpl<mlir::Operation*>& add_ops, bool* changed) { in ApplyOptimization()
470 llvm::SmallVectorImpl<mlir::Operation*>* add_ops, in CollectOptimizationCandidates()
495 llvm::SmallVector<mlir::Operation*, 4> add_ops; in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc661 std::deque<InputAndShape> add_ops; in RewriteOptimizedNodesGroup() local
1071 std::deque<InputAndShape> add_ops(group.inputs.begin(), group.inputs.end()); in RewriteOptimizedNodesGroup() local