Searched defs:scatter_dim (Results 1 – 4 of 4) sorted by relevance
50 mlir::TF::DTensorAllScatterOp all_scatter, int scatter_dim) { in GetScatterGroupAssignment()106 int scatter_dim = scatter_dims[0]; in ApplyOptimization() local
367 mlir::APInt scatter_dim = *scatter_attr.begin(); in LowerReduceScatterOp() local
2616 int64_t scatter_dim; in ReduceScatterShape() local
790 auto scatter_dim = op.scatter_dimension(); in ExportXlaOp() local