- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for CreateSplitOp (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
namespace { constexpr char kNumSplitAttr[] = "num_split"; // Creates a tf::SplitOp that splits 'src_input' into 'num_splits' ways // in 'split_dimension' dimension and returns the split values. mlir::LogicalResult CreateSplitOp(const int num_split, const int split_dimension, const mlir::Location& location, mlir::Value src_input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)