Lines Matching refs:result
22 OpSharding result;
23 result.set_type(OpSharding::Type::OpSharding_Type_REPLICATED);
24 return result;
28 OpSharding result;
29 result.set_type(OpSharding::Type::OpSharding_Type_MAXIMAL);
30 result.add_tile_assignment_dimensions(1);
31 result.add_tile_assignment_devices(device);
32 return result;
37 OpSharding result;
38 result.set_type(OpSharding::Type::OpSharding_Type_OTHER);
39 *result.mutable_tile_shape() = tile_shape;
41 result.add_tile_assignment_dimensions(dim);
44 result.add_tile_assignment_devices(device);
46 return result;
50 OpSharding result;
51 result.set_type(OpSharding::Type::OpSharding_Type_OTHER);
55 *result.mutable_tile_shape() = tile_shape;
57 (*result.mutable_tile_shape()->mutable_dimensions())[0];
59 result.add_tile_assignment_dimensions(num_tiles);
61 result.add_tile_assignment_devices(i);
63 return result;
67 OpSharding result;
68 result.set_type(OpSharding::Type::OpSharding_Type_TUPLE);
70 *result.add_tuple_shardings() = index_to_sharding.second;
72 return result;
