| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | meta_spmd_expander.cc | 172 StatusOr<llvm::DenseMap<int, Layout>> PackSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::PackSPMDExpander
232 StatusOr<llvm::DenseMap<int, Layout>> UnpackSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::UnpackSPMDExpander
312 StatusOr<llvm::DenseMap<int, Layout>> PadSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::PadSPMDExpander
501 StatusOr<llvm::DenseMap<int, Layout>> TileSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TileSPMDExpander
835 ReshapeSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::ReshapeSPMDExpander
926 TransposeSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TransposeSPMDExpander
1111 StatusOr<llvm::DenseMap<int, Layout>> OneHotSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::OneHotSPMDExpander
1240 StatusOr<llvm::DenseMap<int, Layout>> ShapeSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::ShapeSPMDExpander
|
| H A D | slice_spmd_expander.cc | 558 StatusOr<llvm::DenseMap<int, Layout>> SliceSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SliceSPMDExpander
672 StridedSliceSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::StridedSliceSPMDExpander
817 TensorStridedSliceUpdateSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TensorStridedSliceUpdateSPMDExpander
957 StridedSliceGradSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::StridedSliceGradSPMDExpander
|
| H A D | dtensor_op_spmd_expander.cc | 227 RelayoutSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::RelayoutSPMDExpander
382 DTensorSendSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::DTensorSendSPMDExpander
541 DTensorRecvSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::DTensorRecvSPMDExpander
|
| H A D | trivial_spmd_expander.h | 38 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutBackward( in ComputeLayoutBackward() function
55 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutBackward( in ComputeLayoutBackward() function
|
| H A D | control_flow_spmd_expander.cc | 57 WhileRegionSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::WhileRegionSPMDExpander
96 IfRegionSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::IfRegionSPMDExpander
|
| H A D | split_spmd_expander.cc | 113 StatusOr<llvm::DenseMap<int, Layout>> SplitSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SplitSPMDExpander
166 StatusOr<llvm::DenseMap<int, Layout>> SplitVSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SplitVSPMDExpander
|
| H A D | gather_spmd_expander.cc | 238 GatherV2SPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::GatherV2SPMDExpander
506 GatherNdSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::GatherNdSPMDExpander
|
| H A D | softmax_spmd_expander.cc | 360 SoftmaxOpSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SoftmaxOpSPMDExpander
675 SoftmaxLossOpSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SoftmaxLossOpSPMDExpander
|
| H A D | save_restore_spmd_expander.h | 54 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutBackward( in ComputeLayoutBackward() function
|
| H A D | tensorlist_setitem_spmd_expander.cc | 68 TensorListSetItemSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TensorListSetItemSPMDExpander
|
| H A D | tensorlist_getitem_spmd_expander.cc | 55 TensorListGetItemSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TensorListGetItemSPMDExpander
|
| H A D | range_spmd_expander.cc | 47 StatusOr<llvm::DenseMap<int, Layout>> RangeSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::RangeSPMDExpander
|
| H A D | sparse_to_dense_spmd_expander.cc | 50 SparseToDenseSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SparseToDenseSPMDExpander
|
| H A D | tensorlist_reserve_spmd_expander.cc | 71 TensorListReserveSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TensorListReserveSPMDExpander
|
| H A D | identity_n_spmd_expander.cc | 82 IdentityNSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::IdentityNSPMDExpander
|
| H A D | qr_spmd_expander.cc | 104 StatusOr<llvm::DenseMap<int, Layout>> QRSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::QRSPMDExpander
|
| H A D | top_k_spmd_expander.cc | 93 StatusOr<llvm::DenseMap<int, Layout>> TopKSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::TopKSPMDExpander
|
| H A D | fill_spmd_expander.cc | 101 StatusOr<llvm::DenseMap<int, Layout>> FillSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::FillSPMDExpander
|
| H A D | cumsum_spmd_expander.cc | 114 StatusOr<llvm::DenseMap<int, Layout>> CumsumSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::CumsumSPMDExpander
|
| H A D | expanddims_spmd_expander.cc | 102 StatusOr<llvm::DenseMap<int, Layout>> ExpandDimsExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::ExpandDimsExpander
|
| H A D | trivial_spmd_expander.cc | 95 MetadataSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::MetadataSPMDExpander
|
| H A D | segmentation_spmd_expander.cc | 57 UnsortedSegmentSumSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::UnsortedSegmentSumSPMDExpander
|
| H A D | nullary_spmd_expander.cc | 138 NullarySPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::NullarySPMDExpander
|
| H A D | replicated_spmd_expander.cc | 131 ReplicatedOpSPMDExpander::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::ReplicatedOpSPMDExpander
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expander.cc | 162 StatusOr<llvm::DenseMap<int, Layout>> SPMDExpanderBase::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SPMDExpanderBase
168 StatusOr<llvm::DenseMap<int, Layout>> SPMDExpanderBase::ComputeLayoutBackward( in ComputeLayoutBackward() function in tensorflow::dtensor::SPMDExpanderBase
|