Home
last modified time | relevance | path

Searched defs:tile_assignment (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.h53 static HloSharding Tile(const Array<int64>& tile_assignment) { in Tile()
205 const Array<int64>& tile_assignment() const { return tile_assignment_; } in tile_assignment() function
235 explicit HloSharding(const Array<int64>& tile_assignment) in HloSharding()
Dhlo_sharding.cc399 Array<int64> tile_assignment( in FromProto() local
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py64 def tile(cls, tile_assignment): argument
179 def tile(tensor, tile_assignment): argument
/external/tensorflow/tensorflow/compiler/xla/client/
Dsharding_builder.cc36 const TileAssignment& tile_assignment) { in Tile()