Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py60 tile_assignment_dimensions=[1],
88 tile_assignment_dimensions=dims,
120 tile_assignment_dimensions=tile_assignment_dims,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc382 TF_RET_CHECK(!proto.tile_assignment_dimensions().empty()); in FromProto()
387 for (auto dimension : proto.tile_assignment_dimensions()) { in FromProto()
400 std::vector<int64>(proto.tile_assignment_dimensions().begin(), in FromProto()
401 proto.tile_assignment_dimensions().end())); in FromProto()
Dhlo_parser.cc1802 std::vector<int64> tile_assignment_dimensions; in ParseSingleSharding() local
1832 tile_assignment_dimensions.push_back(dim); in ParseSingleSharding()
1877 if (tile_assignment_dimensions.empty()) { in ParseSingleSharding()
1884 for (int64 dim : tile_assignment_dimensions) { in ParseSingleSharding()
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto593 repeated int64 tile_assignment_dimensions = 3; field