Searched refs:create_tuple (Results 1 – 2 of 2) sorted by relevance
423 auto create_tuple = [&](ArrayRef<Value> operands) { in GetValueWithToken() local431 if (!tuple_type) return create_tuple({value, token}); in GetValueWithToken()439 return create_tuple(tuple_operands); in GetValueWithToken()451 return create_tuple(tuple_operands); in GetValueWithToken()
1935 auto create_tuple = b_.AddInstruction( in HandleCustomCall() local1937 create_tuple->set_sharding(HloSharding::Replicate()); in HandleCustomCall()1939 SetPartitionedHlo(hlo, PartitionedHlo(create_tuple, create_tuple->shape(), in HandleCustomCall()