Searched defs:HloSharding (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding.h39 class HloSharding { class in namespace:xla
43 static HloSharding Replicate() { return HloSharding(); }
47 static HloSharding AssignDevice(int64 device_id);
62 static HloSharding Tile(const Shape& tile_shape,
64 return HloSharding(tile_shape, tile_assignment);
69 static HloSharding Tile1D(const Shape& input_shape, int64 num_tiles);
73 static HloSharding Tuple(const ShapeTree<HloSharding>& sub_shardings) {
74 std::vector<HloSharding> flattened_lis
211 HloSharding() function
217 explicit HloSharding(int64 device_id) function
223 HloSharding(const Shape& tile_shape, const Array<int64>& tile_assignment) function
229 HloSharding(const std::vector<HloSharding>& tuple_shardings) function
[all...]

Completed in 186 milliseconds