Searched refs:repack_map (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_test.cc | 4380 absl::flat_hash_map<std::pair<int64, int64>, int64>& repack_map, in FakeMemorySpaceAssignmentRepacker() argument
4384 repack_map_(repack_map), in FakeMemorySpaceAssignmentRepacker()
4528 absl::flat_hash_map<std::pair<int64, int64>, int64> repack_map; in TEST_P() local
4530 repack_map[{2, 0}] = 32; in TEST_P()
4532 repack_map[{3, 32}] = 0; in TEST_P()
4534 FakeMemorySpaceAssignmentRepacker(repack_map); in TEST_P()
4639 absl::flat_hash_map<std::pair<int64, int64>, int64> repack_map; in TEST_P() local
4654 FakeMemorySpaceAssignmentRepacker(repack_map, check_fun); in TEST_P()
4702 absl::flat_hash_map<std::pair<int64, int64>, int64> repack_map; in TEST_P() local
4704 FakeMemorySpaceAssignmentRepacker(repack_map, nullptr, in TEST_P()
|