Home
last modified time | relevance | path

Searched refs:split_sizes_attr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc1590 DenseIntElementsAttr split_sizes_attr; in verify() local
1591 if (!matchPattern(op.size_splits(), m_Constant(&split_sizes_attr))) in verify()
1599 split_sizes_attr.getType().cast<ShapedType>().getNumElements()); in verify()
1601 for (auto dim : llvm::enumerate(split_sizes_attr)) { in verify()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc3505 DenseIntElementsAttr split_sizes_attr; in matchAndRewrite() local
3506 if (!matchPattern(op.size_splits(), m_Constant(&split_sizes_attr))) in matchAndRewrite()
3515 split_sizes_attr.getType().cast<ShapedType>().getNumElements()); in matchAndRewrite()
3516 for (auto dim : llvm::enumerate(split_sizes_attr)) { in matchAndRewrite()