Home
last modified time | relevance | path

Searched defs:group_assignment_attr (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dgroup_assignment_test.cc58 mlir::DenseIntElementsAttr group_assignment_attr = in CreateGroupAssignment() local
73 mlir::DenseIntElementsAttr group_assignment_attr = in CreateGroupAssignment() local
H A Ddtensor_mixed_precision_reduce.cc42 mlir::DenseIntElementsAttr group_assignment_attr; in GetAllReduceGroupSize() local
H A Dgroup_assignment.cc108 const mlir::DenseIntElementsAttr& group_assignment_attr, in FromMLIR()
H A Ddtensor_allreduce_combine_optimization.cc359 mlir::DenseIntElementsAttr group_assignment_attr; in CombineAllReduceOpsOfSameTypeAndGroupAssignment() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc92 mlir::DenseIntElementsAttr group_assignment_attr, int32 key_base, in EmitAllReduceForXla()
212 mlir::DenseIntElementsAttr group_assignment_attr; in LowerAllReduceOpImpl() local
333 mlir::DenseIntElementsAttr group_assignment_attr; in LowerReduceScatterOp() local