Searched refs:GetStridedSliceIntermediateLayout (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | slice_spmd_expander.cc | 267 StatusOr<Layout> GetStridedSliceIntermediateLayout( in GetStridedSliceIntermediateLayout() function
594 GetStridedSliceIntermediateLayout(strided_slice_op, input_layout, in ExpandOp()
660 GetStridedSliceIntermediateLayout( in ComputeLayoutForward()
700 TF_ASSIGN_OR_RETURN(proposed_layout, GetStridedSliceIntermediateLayout( in ComputeLayoutBackward()
732 GetStridedSliceIntermediateLayout(strided_slice_op, input_layout, in ExpandOp()
810 GetStridedSliceIntermediateLayout( in ComputeLayoutForward()
840 GetStridedSliceIntermediateLayout(strided_slice_op, output_layout, in ComputeLayoutBackward()
878 GetStridedSliceIntermediateLayout(strided_slice_grad_op, output_layout, in ExpandOp()
951 GetStridedSliceIntermediateLayout(strided_slice_grad_op, proposed_layout, in ComputeLayoutForward()
982 GetStridedSliceIntermediateLayout(strided_slice_grad_op, output_layout, in ComputeLayoutBackward()
|