- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for kNumSplitAttr (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "xla/client/sharding_builder.h" #include "xla/service/hlo_parser.h" #include "xla/xla_data.pb.h" namespace tensorflow { 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,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)