Home
last modified time | relevance | path

Searched defs:input_layouts (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dsplit_spmd_expander.cc101 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
118 llvm::DenseMap<int, Layout> input_layouts(split_op.getNumOperands()); in ComputeLayoutBackward() local
154 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
171 llvm::DenseMap<int, Layout> input_layouts(split_v_op.getNumOperands()); in ComputeLayoutBackward() local
Drange_spmd_expander.cc40 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
51 llvm::DenseMap<int, Layout> input_layouts; in ComputeLayoutBackward() local
Dmeta_spmd_expander.cc166 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
222 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
290 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
318 llvm::DenseMap<int, Layout> input_layouts(op->getNumOperands()); in ComputeLayoutBackward() local
454 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
518 llvm::DenseMap<int, Layout> input_layouts(op->getNumOperands()); in ComputeLayoutBackward() local
803 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
906 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
933 llvm::DenseMap<int, Layout> input_layouts(transpose->getNumOperands()); in ComputeLayoutBackward() local
1084 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
[all …]
Dtrivial_spmd_expander.h34 const llvm::DenseMap<int, Layout>& input_layouts) override { in ComputeLayoutForward()
51 const llvm::DenseMap<int, Layout>& input_layouts) override { in ComputeLayoutForward()
Dslice_spmd_expander.cc544 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
563 llvm::DenseMap<int, Layout> input_layouts(slice_op.getNumOperands()); in ComputeLayoutBackward() local
645 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
684 llvm::DenseMap<int, Layout> input_layouts(strided_slice_op.getNumOperands()); in ComputeLayoutBackward() local
778 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
829 llvm::DenseMap<int, Layout> input_layouts(strided_slice_op.getNumOperands()); in ComputeLayoutBackward() local
929 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
969 llvm::DenseMap<int, Layout> input_layouts( in ComputeLayoutBackward() local
Dscatter_spmd_expander.cc155 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in TensorScatterOpComputeLayoutForward()
197 llvm::DenseMap<int, Layout> input_layouts(scatter_op.getNumOperands()); in TensorScatterOpComputeLayoutBackward() local
246 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dtrivial_spmd_expander.cc81 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
99 llvm::DenseMap<int, Layout> input_layouts(op->getNumOperands()); in ComputeLayoutBackward() local
Dcontrol_flow_spmd_expander.cc50 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
90 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dreplicated_spmd_expander.cc118 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
135 llvm::DenseMap<int, Layout> input_layouts(op->getNumOperands()); in ComputeLayoutBackward() local
Delementwise_spmd_expander.cc124 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
150 llvm::DenseMap<int, Layout> input_layouts; in ComputeLayoutBackward() local
Dbias_add_spmd_expander.cc117 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
153 llvm::DenseMap<int, Layout> input_layouts; in ComputeLayoutBackward() local
Dresource_spmd_expander.cc220 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
248 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts, in ComputeLayoutBackward()
Deinsum_spmd_expander.cc44 std::vector<Layout> input_layouts(op->getNumOperands()); in ExpandOp() local
250 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
350 llvm::DenseMap<int, Layout> input_layouts(num_inputs); in ComputeLayoutBackward() local
388 const std::vector<Layout>& input_layouts, mlir::Operation* op, in MaybeRelayoutInputs()
Dsave_restore_spmd_expander.h48 const llvm::DenseMap<int, Layout>& input_layouts) override { in ComputeLayoutForward()
Dtensorlist_setitem_spmd_expander.cc55 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dtensorlist_getitem_spmd_expander.cc46 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Ddtensor_op_spmd_expander.cc222 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
377 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
530 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dsparse_to_dense_spmd_expander.cc44 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dgather_spmd_expander.cc169 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
243 llvm::DenseMap<int, Layout> input_layouts(gather_op.getNumOperands()); in ComputeLayoutBackward() local
469 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dtensorlist_reserve_spmd_expander.cc63 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Didentity_n_spmd_expander.cc77 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dqr_spmd_expander.cc91 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
Dfill_spmd_expander.cc93 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expander.cc151 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts) { in ComputeLayoutForward()
157 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts, in ComputeLayoutForward()
169 mlir::Operation* op, const llvm::DenseMap<int, Layout>& input_layouts, in ComputeLayoutBackward()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_ops.cc92 std::vector<std::string> input_layouts; in __anonac88f2010202() local

12