Home
last modified time | relevance | path

Searched refs:ClearSharding (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py752 def ClearSharding(self): member in ComputationBuilder
757 self._builder.ClearSharding()
Dxla.cc807 .def("ClearSharding", &XlaBuilder::ClearSharding); in PYBIND11_MODULE()
Dxla_client_test.py2049 c.ClearSharding()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h192 void ClearSharding() { sharding_ = absl::nullopt; } in ClearSharding() function
1071 builder_->ClearSharding(); in SetSharding()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc796 builder.ClearSharding(); in XLA_TEST_F()