tensorflow/tensorflow/tensorflow/dtensor/mlir/expansions/ Source Tree

../
argmax_spmd_expander.cc
argmax_spmd_expander.h ArgMaxSPMDExpander
bias_add_spmd_expander.cc
bias_add_spmd_expander.h BiasAddExpander
broadcast_to_spmd_expander.cc
broadcast_to_spmd_expander.h BroadcastToSPMDExpander
concat_spmd_expander.cc
concat_spmd_expander.h ConcatSPMDExpander
control_flow_spmd_expander.cc
control_flow_spmd_expander.h IfRegionSPMDExpander WhileRegionSPMDExpander
conv_spmd_expander.cc
conv_spmd_expander.h ConvSPMDExpander
cumsum_spmd_expander.cc
cumsum_spmd_expander.h CumsumSPMDExpander
dataparallel_spmd_expander.cc
dataparallel_spmd_expander.h DataparallelSPMDExpander
disable_copy_on_read_spmd_expander.cc
disable_copy_on_read_spmd_expander.h DisableCopyOnReadSPMDExpander
dtensor_op_spmd_expander.cc
dtensor_op_spmd_expander.h DTensorRecvSPMDExpander DTensorSendSPMDExpander RelayoutSPMDExpander
einsum_spmd_expander.cc
einsum_spmd_expander.h EinsumSPMDExpander
elementwise_spmd_expander.cc
elementwise_spmd_expander.h ElementwiseSPMDExpander
expanddims_spmd_expander.cc
expanddims_spmd_expander.h ExpandDimsExpander
fill_spmd_expander.cc
fill_spmd_expander.h FillSPMDExpander
gather_spmd_expander.cc
gather_spmd_expander.h GatherNdSPMDExpander GatherV2SPMDExpander
identity_n_spmd_expander.cc
identity_n_spmd_expander.h IdentityNSPMDExpander
in_top_k_spmd_expander.cc
in_top_k_spmd_expander.h InTopKSPMDExpander
io_op_spmd_expander.cc
io_op_spmd_expander.h IOOpSPMDExpander
matmul_spmd_expander.cc
matmul_spmd_expander.h MatMulSPMDExpander
meta_spmd_expander.cc
meta_spmd_expander.h OneHotSPMDExpander PackSPMDExpander PadSPMDExpander ReshapeSPMDExpander ShapeSPMDExpander TileSPMDExpander TransposeSPMDExpander UnpackSPMDExpander
nullary_spmd_expander.cc
nullary_spmd_expander.h NullarySPMDExpander
qr_spmd_expander.cc
qr_spmd_expander.h QRSPMDExpander
random_op_spmd_expander.cc
random_op_spmd_expander.h RandomOpSPMDExpander
range_spmd_expander.cc
range_spmd_expander.h RangeSPMDExpander
reduce_spmd_expander.cc
reduce_spmd_expander.h ReduceSPMDExpander
replicated_spmd_expander.cc
replicated_spmd_expander.h ReplicatedOpSPMDExpander
resource_spmd_expander.cc
resource_spmd_expander.h ResourceSPMDExpander
save_restore_spmd_expander.cc
save_restore_spmd_expander.h DTensorShardPrefixSPMDExpander SaveRestoreSPMDExpander
scatter_spmd_expander.cc
scatter_spmd_expander.h TensorScatterOpSPMDExpander
segmentation_spmd_expander.cc
segmentation_spmd_expander.h UnsortedSegmentSumSPMDExpander
slice_spmd_expander.cc
slice_spmd_expander.h SliceSPMDExpander StridedSliceGradSPMDExpander StridedSliceSPMDExpander TensorStridedSliceUpdateSPMDExpander
softmax_spmd_expander.cc
softmax_spmd_expander.h SoftmaxLossOpSPMDExpander SoftmaxOpSPMDExpander
sparse_to_dense_spmd_expander.cc
sparse_to_dense_spmd_expander.h SparseToDenseSPMDExpander
split_spmd_expander.cc
split_spmd_expander.h SplitSPMDExpander SplitVSPMDExpander
squeeze_spmd_expander.cc
squeeze_spmd_expander.h SqueezeSPMDExpander
tensorlist_getitem_spmd_expander.cc
tensorlist_getitem_spmd_expander.h TensorListGetItemSPMDExpander
tensorlist_reserve_spmd_expander.cc
tensorlist_reserve_spmd_expander.h TensorListReserveSPMDExpander
tensorlist_setitem_spmd_expander.cc
tensorlist_setitem_spmd_expander.h TensorListSetItemSPMDExpander
top_k_spmd_expander.cc
top_k_spmd_expander.h TopKSPMDExpander
trivial_spmd_expander.cc
trivial_spmd_expander.h MetadataSPMDExpander NoOpSPMDExpander TerminatorSPMDExpander

Browse the source code of tensorflow/tensorflow/tensorflow/dtensor/mlir/expansions/ online