Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched defs:tile_shape (Results 1 - 4 of 4) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/
hlo_sharding_test.cc	43 Shape tile_shape = ShapeUtil::MakeShape(U32, {4}); local 83 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 85 HloSharding::Tile(tile_shape, MakeArray({2, 2}, {0, 0, 2, 3})); 92 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 94 HloSharding::Tile(tile_shape, MakeArray({2, 2}, {0, 1, 2, 3})); 101 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 103 HloSharding::Tile(tile_shape, MakeArray({2, 2}, {0, 1, 2, 3})); 111 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 113 HloSharding::Tile(tile_shape, MakeArray({2, 2}, {0, 1, 2, 3})); 120 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}) local 200 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 209 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local 218 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3}); local     [all...]
hlo_sharding.h	50 // `tile_shape`. Each tile is assigned to one device, which is specified by 62 static HloSharding Tile(const Shape& tile_shape, 64 return HloSharding(tile_shape, tile_assignment); 205 const Shape& tile_shape() const { return tile_shape_; } function 223 HloSharding(const Shape& tile_shape, const Array<int64>& tile_assignment) 227 tile_shape_(tile_shape),
hlo_sharding.cc	33 Shape tile_shape = input_shape; local 34 auto& tile_dimension = (*tile_shape.mutable_dimensions())[0]; 38 return HloSharding(tile_shape, assignment); 316 return HloSharding(proto.tile_shape(), tile_assignment);
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc	1131 Shape tile_shape; local     [all...]

Completed in 204 milliseconds