| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | meta_spmd_expander.cc | 165 StatusOr<llvm::DenseMap<int, Layout>> PackSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::PackSPMDExpander
221 StatusOr<llvm::DenseMap<int, Layout>> UnpackSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::UnpackSPMDExpander
289 StatusOr<llvm::DenseMap<int, Layout>> PadSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::PadSPMDExpander
453 StatusOr<llvm::DenseMap<int, Layout>> TileSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TileSPMDExpander
802 StatusOr<llvm::DenseMap<int, Layout>> ReshapeSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::ReshapeSPMDExpander
905 TransposeSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TransposeSPMDExpander
1083 StatusOr<llvm::DenseMap<int, Layout>> OneHotSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::OneHotSPMDExpander
1231 StatusOr<llvm::DenseMap<int, Layout>> ShapeSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::ShapeSPMDExpander
|
| H A D | slice_spmd_expander.cc | 543 StatusOr<llvm::DenseMap<int, Layout>> SliceSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SliceSPMDExpander
644 StridedSliceSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::StridedSliceSPMDExpander
777 TensorStridedSliceUpdateSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TensorStridedSliceUpdateSPMDExpander
928 StridedSliceGradSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::StridedSliceGradSPMDExpander
|
| H A D | dtensor_op_spmd_expander.cc | 221 RelayoutSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::RelayoutSPMDExpander
376 DTensorSendSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::DTensorSendSPMDExpander
529 DTensorRecvSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::DTensorRecvSPMDExpander
|
| H A D | trivial_spmd_expander.h | 32 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutForward( in ComputeLayoutForward() function
49 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutForward( in ComputeLayoutForward() function
|
| H A D | control_flow_spmd_expander.cc | 49 WhileRegionSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::WhileRegionSPMDExpander
89 IfRegionSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::IfRegionSPMDExpander
|
| H A D | split_spmd_expander.cc | 100 StatusOr<llvm::DenseMap<int, Layout>> SplitSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SplitSPMDExpander
153 StatusOr<llvm::DenseMap<int, Layout>> SplitVSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SplitVSPMDExpander
|
| H A D | gather_spmd_expander.cc | 168 GatherV2SPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::GatherV2SPMDExpander
468 GatherNdSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::GatherNdSPMDExpander
|
| H A D | softmax_spmd_expander.cc | 353 SoftmaxOpSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SoftmaxOpSPMDExpander
622 SoftmaxLossOpSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SoftmaxLossOpSPMDExpander
|
| H A D | save_restore_spmd_expander.h | 46 StatusOr<llvm::DenseMap<int, Layout>> ComputeLayoutForward( in ComputeLayoutForward() function
|
| H A D | tensorlist_setitem_spmd_expander.cc | 54 TensorListSetItemSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TensorListSetItemSPMDExpander
|
| H A D | tensorlist_getitem_spmd_expander.cc | 45 TensorListGetItemSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TensorListGetItemSPMDExpander
|
| H A D | range_spmd_expander.cc | 39 StatusOr<llvm::DenseMap<int, Layout>> RangeSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::RangeSPMDExpander
|
| H A D | sparse_to_dense_spmd_expander.cc | 43 SparseToDenseSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SparseToDenseSPMDExpander
|
| H A D | tensorlist_reserve_spmd_expander.cc | 62 TensorListReserveSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TensorListReserveSPMDExpander
|
| H A D | identity_n_spmd_expander.cc | 76 IdentityNSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::IdentityNSPMDExpander
|
| H A D | qr_spmd_expander.cc | 90 StatusOr<llvm::DenseMap<int, Layout>> QRSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::QRSPMDExpander
|
| H A D | top_k_spmd_expander.cc | 78 StatusOr<llvm::DenseMap<int, Layout>> TopKSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::TopKSPMDExpander
|
| H A D | fill_spmd_expander.cc | 92 StatusOr<llvm::DenseMap<int, Layout>> FillSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::FillSPMDExpander
|
| H A D | cumsum_spmd_expander.cc | 100 StatusOr<llvm::DenseMap<int, Layout>> CumsumSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::CumsumSPMDExpander
|
| H A D | expanddims_spmd_expander.cc | 74 StatusOr<llvm::DenseMap<int, Layout>> ExpandDimsExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::ExpandDimsExpander
|
| H A D | trivial_spmd_expander.cc | 80 MetadataSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::MetadataSPMDExpander
|
| H A D | segmentation_spmd_expander.cc | 34 UnsortedSegmentSumSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::UnsortedSegmentSumSPMDExpander
|
| H A D | nullary_spmd_expander.cc | 113 StatusOr<llvm::DenseMap<int, Layout>> NullarySPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::NullarySPMDExpander
|
| H A D | replicated_spmd_expander.cc | 117 ReplicatedOpSPMDExpander::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::ReplicatedOpSPMDExpander
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | spmd_expander.cc | 150 StatusOr<llvm::DenseMap<int, Layout>> SPMDExpanderBase::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SPMDExpanderBase
156 StatusOr<llvm::DenseMap<int, Layout>> SPMDExpanderBase::ComputeLayoutForward( in ComputeLayoutForward() function in tensorflow::dtensor::SPMDExpanderBase
|