Home
last modified time | relevance | path

Searched defs:CreateSplitOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.cc45 mlir::LogicalResult CreateSplitOp(const int num_split, in CreateSplitOp() function
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.cc97 Status CreateSplitOp(const int num_split, const int split_dimension, in CreateSplitOp() function