- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getSizeSplits (0.29 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// SplitVOp //===----------------------------------------------------------------------===// LogicalResult SplitVOp::verify() { SplitVOp op = *this; auto split_sizes_type = op.getSizeSplits().getType().dyn_cast<RankedTensorType>(); if (!split_sizes_type) return success(); if (split_sizes_type.getRank() != 1 || (!ShapedType::isDynamic(split_sizes_type.getDimSize(0)) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
// If 'size_splits' is not a constant, there are no other checks. ElementsAttr size_splits_attr; if (!matchPattern(op.getSizeSplits(), m_Constant(&size_splits_attr))) return success(); if (size_splits_attr.getNumElements() != num_splits) { auto size_splits_type = op.getSizeSplits().getType().cast<RankedTensorType>(); RankedTensorType expected_size_splits_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)