Home
last modified time | relevance | path

Searched defs:ExpandOp (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dmeta_spmd_expander.cc131 StatusOr<mlir::Operation*> PackSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::PackSPMDExpander
183 StatusOr<mlir::Operation*> UnpackSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::UnpackSPMDExpander
270 StatusOr<mlir::Operation*> PadSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::PadSPMDExpander
362 StatusOr<mlir::Operation*> TileSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::TileSPMDExpander
660 StatusOr<mlir::Operation*> ReshapeSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::ReshapeSPMDExpander
867 StatusOr<mlir::Operation*> TransposeSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TransposeSPMDExpander
990 StatusOr<mlir::Operation*> OneHotSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::OneHotSPMDExpander
1144 StatusOr<mlir::Operation*> ShapeSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::ShapeSPMDExpander
H A Dslice_spmd_expander.cc449 StatusOr<mlir::Operation*> SliceSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::SliceSPMDExpander
580 StatusOr<mlir::Operation*> StridedSliceSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::StridedSliceSPMDExpander
709 StatusOr<mlir::Operation*> TensorStridedSliceUpdateSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TensorStridedSliceUpdateSPMDExpander
859 StatusOr<mlir::Operation*> StridedSliceGradSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::StridedSliceGradSPMDExpander
H A Ddtensor_op_spmd_expander.cc174 StatusOr<mlir::Operation*> RelayoutSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::RelayoutSPMDExpander
265 StatusOr<mlir::Operation*> DTensorSendSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::DTensorSendSPMDExpander
387 StatusOr<mlir::Operation*> DTensorRecvSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::DTensorRecvSPMDExpander
H A Dtrivial_spmd_expander.cc34 StatusOr<mlir::Operation*> TerminatorSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TerminatorSPMDExpander
49 StatusOr<mlir::Operation*> MetadataSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::MetadataSPMDExpander
H A Dcontrol_flow_spmd_expander.cc21 StatusOr<mlir::Operation*> WhileRegionSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::WhileRegionSPMDExpander
64 StatusOr<mlir::Operation*> IfRegionSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::IfRegionSPMDExpander
H A Dsplit_spmd_expander.cc84 StatusOr<mlir::Operation*> SplitSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::SplitSPMDExpander
137 StatusOr<mlir::Operation*> SplitVSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::SplitVSPMDExpander
H A Dgather_spmd_expander.cc35 StatusOr<mlir::Operation*> GatherV2SPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::GatherV2SPMDExpander
370 StatusOr<mlir::Operation*> GatherNdSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::GatherNdSPMDExpander
H A Dsoftmax_spmd_expander.cc322 StatusOr<mlir::Operation*> SoftmaxOpSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::SoftmaxOpSPMDExpander
511 StatusOr<mlir::Operation*> SoftmaxLossOpSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::SoftmaxLossOpSPMDExpander
H A Dsave_restore_spmd_expander.h42 StatusOr<mlir::Operation*> ExpandOp(mlir::Operation* op) override { in ExpandOp() function
H A Dtensorlist_setitem_spmd_expander.cc26 StatusOr<mlir::Operation*> TensorListSetItemSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TensorListSetItemSPMDExpander
H A Dtensorlist_getitem_spmd_expander.cc25 StatusOr<mlir::Operation*> TensorListGetItemSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TensorListGetItemSPMDExpander
H A Drange_spmd_expander.cc26 StatusOr<mlir::Operation*> RangeSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::RangeSPMDExpander
H A Dsparse_to_dense_spmd_expander.cc25 StatusOr<mlir::Operation*> SparseToDenseSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::SparseToDenseSPMDExpander
H A Dtrivial_spmd_expander.h28 StatusOr<mlir::Operation*> ExpandOp(mlir::Operation* op) override { in ExpandOp() function
H A Dtensorlist_reserve_spmd_expander.cc29 StatusOr<mlir::Operation*> TensorListReserveSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::TensorListReserveSPMDExpander
H A Didentity_n_spmd_expander.cc26 StatusOr<mlir::Operation*> IdentityNSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::IdentityNSPMDExpander
H A Dqr_spmd_expander.cc42 StatusOr<mlir::Operation*> QRSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::QRSPMDExpander
H A Dtop_k_spmd_expander.cc43 StatusOr<mlir::Operation*> TopKSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::TopKSPMDExpander
H A Dfill_spmd_expander.cc32 StatusOr<mlir::Operation*> FillSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::FillSPMDExpander
H A Dcumsum_spmd_expander.cc61 StatusOr<mlir::Operation*> CumsumSPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::CumsumSPMDExpander
H A Dexpanddims_spmd_expander.cc30 StatusOr<mlir::Operation*> ExpandDimsExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::ExpandDimsExpander
H A Dsegmentation_spmd_expander.cc82 StatusOr<mlir::Operation*> UnsortedSegmentSumSPMDExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::UnsortedSegmentSumSPMDExpander
H A Dnullary_spmd_expander.cc40 StatusOr<mlir::Operation*> NullarySPMDExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::NullarySPMDExpander
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
H A Dmatmul_sparse_expander.cc25 StatusOr<mlir::Operation*> MatMulSparseExpander::ExpandOp(mlir::Operation* op) { in ExpandOp() function in tensorflow::dtensor::MatMulSparseExpander
H A Ddynamic_enqueue_sparse_expander.cc63 StatusOr<mlir::Operation*> DynamicEnqueueSparseExpander::ExpandOp( in ExpandOp() function in tensorflow::dtensor::DynamicEnqueueSparseExpander

12