Searched defs:OpOperand (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | mesh_propagation.cc | 117 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in ExtractMeshFromOperand()
189 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in InferMeshFromInputs()
191 llvm::SmallVector<mlir::OpOperand*, 8>* inputs_with_inferred_mesh) { in InferMeshFromInputs()
270 llvm::SmallVector<mlir::OpOperand*, 8>* consumers_with_mesh) { in InferMeshFromConsumers()
331 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in InferFunctionDefaultMesh()
384 const llvm::SmallVector<mlir::OpOperand*, 8>& input_values_from_mesh, in AnnotateFunctionArgumentsWithMeshInformation()
396 const llvm::SmallVector<mlir::OpOperand*, 8>& return_values_from_mesh, in AnnotateFunctionReturnValuesWithMeshInformation()
513 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in PropagateDefaultMeshToUnAssignedClusters()
550 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in PropagateMeshFromInputs()
610 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in PropagateMeshFromConsumers()
[all …]
|
| H A D | layout_propagation_v2.cc | 191 const mlir::DenseMap<mlir::OpOperand*, Layout>& consumers) { in MergeLayouts()
317 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in InsertInitialLayoutsFromComputeLayout()
319 llvm::DenseMap<mlir::Value, mlir::DenseMap<mlir::OpOperand*, Layout>>& in InsertInitialLayoutsFromComputeLayout()
404 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in InsertInitialLayouts()
405 llvm::DenseMap<mlir::Value, mlir::DenseMap<mlir::OpOperand*, Layout>>& in InsertInitialLayouts()
426 llvm::DenseMap<mlir::Value, mlir::DenseMap<mlir::OpOperand*, Layout>>& in MergeAndGetUpdatedLayouts()
562 const llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in UpdateLayoutsForOp()
565 llvm::DenseMap<mlir::Value, mlir::DenseMap<mlir::OpOperand*, Layout>>& in UpdateLayoutsForOp()
1246 llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>>& producers, in FindRoot()
1295 llvm::DenseMap<mlir::OpOperand*, std::vector<mlir::Value>> producers, in FindRootsAndEmitError()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_fusion.cc | 78 OpOperand &) { in ControlElementwiseOpsFusion() argument
|