Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc775 ElementsAttr dims_elems; in matchAndRewrite() local
776 if (!matchPattern(tf_fill_op.dims(), m_Constant(&dims_elems))) in matchAndRewrite()
780 for (int i = 0; i < dims_elems.getNumElements(); i++) { in matchAndRewrite()
781 dims_vals.push_back(dims_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
Dlegalize_tfl.cc1693 ElementsAttr dims_elems; in matchAndRewrite() local
1694 if (!matchPattern(tfl_fill_op.dims(), m_Constant(&dims_elems))) in matchAndRewrite()
1698 for (int i = 0; i < dims_elems.getNumElements(); i++) { in matchAndRewrite()
1699 dims_vals.push_back(dims_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()