Home
last modified time | relevance | path

Searched defs:arg_shardings (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc100 std::map<int, xla::OpSharding> arg_shardings; in ComputeArgAndRetvalShardings() local
167 const std::map<int, xla::OpSharding>& arg_shardings, in BuildComputation()
956 const std::map<int, xla::OpSharding>& arg_shardings, in BuildArguments()
1327 std::map<int, xla::OpSharding> arg_shardings; in CompileGraph() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc413 llvm::SmallVectorImpl<absl::optional<xla::OpSharding>>* arg_shardings, in ExtractShardingsFromFunction()
1404 llvm::SmallVector<absl::optional<xla::OpSharding>, 4> arg_shardings; in RunOnFunction() local
1481 llvm::ArrayRef<absl::optional<xla::OpSharding>> arg_shardings, in SetEntryTupleShardings()
1510 llvm::ArrayRef<absl::optional<xla::OpSharding>> arg_shardings, in LowerBasicBlockAsFunction()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc2793 const std::vector<xla::OpSharding>& arg_shardings, in CreateOrGetPerHostVariableCopy()
2892 const std::vector<xla::OpSharding>& arg_shardings, in BuildExecuteNodes()