Home
last modified time | relevance | path

Searched refs:Tile1D (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dsharding_builder.h51 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles);
Dsharding_builder.cc49 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles) { in Tile1D() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.h59 static HloSharding Tile1D(const Shape& input_shape, int64 num_tiles);
Dhlo_sharding.cc33 HloSharding HloSharding::Tile1D(const Shape& input_shape, int64 num_tiles) { in Tile1D() function in xla::HloSharding