Searched defs:operand_layouts (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | replicated_spmd_expander.cc | 45 mlir::Operation* op, const std::vector<Layout>& operand_layouts, in ReplicatedRelayoutOperandsAndOutputs()
|
| D | dataparallel_spmd_expander.cc | 149 const std::vector<Layout>& operand_layouts, in IntermediateBatchLayout()
207 mlir::Operation* op, const std::vector<Layout>& operand_layouts, in RelayoutOperandsAndOutputs()
|
| D | concat_spmd_expander.cc | 152 llvm::DenseMap<int, Layout> operand_layouts; in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.cc | 619 std::optional<std::vector<Shape>> operand_layouts, bool has_side_effect) { in EmitPythonCallbackFromDescriptor()
691 std::optional<std::vector<Shape>> operand_layouts, bool has_side_effect) { in EmitPythonCallback()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | spmd_expander_common.cc | 278 const llvm::DenseMap<int, Layout>& operand_layouts, mlir::Operation* op) { in GetMergedOperandLayout()
|