Searched refs:create_tuple (Results 1 – 3 of 3) sorted by relevance
66 auto create_tuple(const std::vector<IValue>& arguments) { in create_tuple() function73 return create_tuple<1>(ivalues); in constructTuple()75 return create_tuple<2>(ivalues); in constructTuple()77 return create_tuple<3>(ivalues); in constructTuple()79 return create_tuple<4>(ivalues); in constructTuple()81 return create_tuple<5>(ivalues); in constructTuple()83 return create_tuple<6>(ivalues); in constructTuple()85 return create_tuple<7>(ivalues); in constructTuple()87 return create_tuple<8>(ivalues); in constructTuple()89 return create_tuple<9>(ivalues); in constructTuple()[all …]
229 auto create_tuple = b_.AddInstruction( in HandleCustomCallTopK() local231 create_tuple->set_sharding( in HandleCustomCallTopK()232 replicated_sharding.GetTupleSharding(create_tuple->shape()).ValueOrDie()); in HandleCustomCallTopK()234 hlo, PartitionedHlo(create_tuple, hlo->shape(), MakePartitioningState()) in HandleCustomCallTopK()
430 auto create_tuple = [&](ArrayRef<Value> operands) { in GetValueWithToken() local438 if (!tuple_type) return {create_tuple({value, token})}; in GetValueWithToken()446 return {create_tuple(tuple_operands)}; in GetValueWithToken()458 return {create_tuple(tuple_operands)}; in GetValueWithToken()