Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1864 DenseIntElementsAttr sparse_begin_attr, sparse_end_attr, sparse_strides_attr; in GetSlicedBoundRanges() local
1867 !matchPattern(strides(), m_Constant(&sparse_strides_attr))) in GetSlicedBoundRanges()
1880 for (const APInt &stride : sparse_strides_attr) in GetSlicedBoundRanges()
2031 DenseIntElementsAttr sparse_begin_attr, sparse_end_attr, sparse_strides_attr; in GetSlicedShapeAndBoundRanges() local
2035 !matchPattern(strides(), m_Constant(&sparse_strides_attr))) in GetSlicedShapeAndBoundRanges()
2051 for (const APInt &stride : sparse_strides_attr) in GetSlicedShapeAndBoundRanges()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3167 DenseIntElementsAttr sparse_strides_attr; in rewriteWithUnknownBegin() local
3168 if (!matchPattern(op.strides(), m_Constant(&sparse_strides_attr))) { in rewriteWithUnknownBegin()
3175 for (const APInt &stride : sparse_strides_attr) { in rewriteWithUnknownBegin()