Home
last modified time | relevance | path

Searched defs:attr_b (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_allreduce_scatter_optimization.cc42 mlir::DenseIntElementsAttr attr_b) { in same_group_assignments()
Ddtensor_allreduce_combine_optimization.cc496 mlir::DenseIntElementsAttr attr_b; in same_group_assignments() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2420 const string attr_b = IsAnyBatchMatMul(*node) ? "adj_y" : "transpose_b"; in TrySimplify() local