Searched refs:NoSharding (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | pmap_lib.cc | 48 cpp_sharding.push_back(NoSharding()); in PyShardingToCpp()
68 if (absl::holds_alternative<NoSharding>(value)) { in CppShardingToPy()
342 py::class_<NoSharding> no_sharding(pmap_lib, "NoSharding"); in BuildPmapSubmodule()
345 [](const NoSharding& chuncked) { return "NoSharding()"; }) in BuildPmapSubmodule()
346 .def("__eq__", [](const NoSharding& self, py::object obj) { in BuildPmapSubmodule()
347 return py::isinstance<NoSharding>(obj); in BuildPmapSubmodule()
|
| D | pmap_lib.h | 52 struct NoSharding {}; struct
85 using AvalDimSharding = absl::variant<NoSharding, Chunked, Unstacked>;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers_test.cc | 179 EXPECT_THAT(p0.get(), op::NoSharding()); in TEST_F()
182 EXPECT_THAT(p1.get(), ::testing::Not(op::NoSharding())); in TEST_F()
194 EXPECT_THAT(Explain(p1.get(), op::NoSharding()), in TEST_F()
|
| D | hlo_matchers.h | 415 inline ::testing::Matcher<const ::xla::HloInstruction*> NoSharding() { in NoSharding() function
|
| D | sharding_propagation_test.cc | 1248 EXPECT_THAT(instruction, op::NoSharding()); in TEST_P()
1608 EXPECT_THAT(crs_f_none, op::NoSharding()); in TEST_P()
2424 EXPECT_THAT(c0, op::NoSharding()); in TEST_P()
2460 EXPECT_THAT(instruction, op::NoSharding()); in TEST_P()
|