- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for VerifySplitInputAndSplitDim (1.03 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// Verifies the input and split dimension operands for tf.Split/tf.SplitV. // Writes the split dimension's index (adjusted with input rank) via `dim_index` // if it's a constant. template <class Op> LogicalResult VerifySplitInputAndSplitDim(Op op, std::optional<int64_t> *dim_index) { *dim_index = std::nullopt; Value split_dim = op.getSplitDim();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)