Searched defs:merged_layouts (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | layout_propagation_v2.cc | 428 llvm::DenseMap<mlir::Value, Layout>& merged_layouts) { in MergeAndGetUpdatedLayouts()
529 const llvm::DenseMap<mlir::Value, Layout>& merged_layouts, in MergeProducerLayouts()
563 const llvm::DenseMap<mlir::Value, Layout>& merged_layouts, in UpdateLayoutsForOp()
681 const llvm::DenseMap<mlir::Value, Layout>& merged_layouts) { in InsertDTensorLayoutOps()
741 const llvm::DenseMap<mlir::Value, Layout>& merged_layouts) in LayoutPrinter()
967 const llvm::DenseMap<mlir::Value, Layout>& merged_layouts, in LogLayoutsAndOps()
1340 llvm::DenseMap<mlir::Value, Layout>& merged_layouts, mlir::ModuleOp& module, in RunOneIteration()
1466 llvm::DenseMap<mlir::Value, Layout> merged_layouts; in runOnOperation() local
|