Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_compiler.cc406 .def("clear_sharding", &XlaBuilder::ClearSharding) in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc1340 builder->ClearSharding(); in Lower()
1533 builder->ClearSharding(); in LowerBasicBlockAsFunction()
1542 builder->ClearSharding(); in LowerBasicBlockAsFunction()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc796 builder.ClearSharding(); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h227 void ClearSharding() { sharding_ = absl::nullopt; } in ClearSharding() function
1499 builder_->ClearSharding(); in SetSharding()