Searched defs:key_base (Results 1 – 2 of 2) sorted by relevance
92 mlir::DenseIntElementsAttr group_assignment_attr, int32 key_base, in EmitAllReduceForXla()120 const mlir::DenseIntElementsAttr& group_assignment, int32 key_base, in EmitCollectiveReduce()223 int32 key_base = tf_collective_key_base++; in LowerAllReduceOpImpl() local
178 def _testWhile(self, num_vars, num_iterations, key_base): argument