Searched refs:tile_assignment_dimensions (Results 1 – 4 of 4) sorted by relevance
60 tile_assignment_dimensions=[1],88 tile_assignment_dimensions=dims,120 tile_assignment_dimensions=tile_assignment_dims,
382 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()
1802 std::vector<int64> tile_assignment_dimensions; in ParseSingleSharding() local1832 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()
593 repeated int64 tile_assignment_dimensions = 3; field