| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | collectives.cc | 53 llvm::SmallPtrSet<mlir::Operation*, 4>* newly_created_ops) { in EmitAllGather()
108 llvm::SmallPtrSet<mlir::Operation*, 4>* newly_created_ops) { in EmitAllScatter()
151 llvm::SmallPtrSet<mlir::Operation*, 4>* newly_created_ops) { in EmitDenseToSparseToDense()
201 llvm::SmallPtrSet<mlir::Operation*, 4>* newly_created_ops) { in EmitRelayout()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | slice_spmd_expander.cc | 483 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
631 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
764 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
914 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | tensorlist_getitem_spmd_expander.cc | 34 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | identity_n_spmd_expander.cc | 30 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | qr_spmd_expander.cc | 61 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | cumsum_spmd_expander.cc | 91 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | expanddims_spmd_expander.cc | 62 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | nullary_spmd_expander.cc | 84 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | replicated_spmd_expander.cc | 61 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ReplicatedRelayoutOperandsAndOutputs() local
|
| H A D | elementwise_spmd_expander.cc | 83 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | broadcast_to_spmd_expander.cc | 112 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | bias_add_spmd_expander.cc | 108 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | resource_spmd_expander.cc | 139 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | reduce_spmd_expander.cc | 247 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | dataparallel_spmd_expander.cc | 233 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in RelayoutOperandsAndOutputs() local
|
| H A D | dtensor_op_spmd_expander.cc | 516 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in ExpandOp() local
|
| H A D | softmax_spmd_expander.cc | 474 llvm::SmallPtrSet<mlir::Operation*, 4> newly_created_ops; in MaybeRelayoutOutputs() local
|