Searched defs:producer_request (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | layout_propagation_v2.cc | 290 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in InsertLayoutsForDTensorLayout()
318 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in InsertInitialLayoutsFromComputeLayout()
407 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in InsertInitialLayouts()
425 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in MergeAndGetUpdatedLayouts()
564 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in UpdateLayoutsForOp()
1335 llvm::DenseMap<mlir::Value, absl::optional<Layout>>& producer_request, in RunOneIteration()
1431 llvm::DenseMap<mlir::Value, absl::optional<Layout>> producer_request; in runOnOperation() local
|