Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc778 SmallVector<int64_t, 4> dims_vals; in matchAndRewrite() local
781 dims_vals.push_back(dims_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
782 total_size *= dims_vals[i]; in matchAndRewrite()
790 ArrayRef<int64_t>(dims_vals), value_elem.getType().getElementType()); in matchAndRewrite()
Dlegalize_tfl.cc1696 SmallVector<int64_t, 4> dims_vals; in matchAndRewrite() local
1699 dims_vals.push_back(dims_elems.getValue<IntegerAttr>(i).getInt()); in matchAndRewrite()
1700 total_size *= dims_vals[i]; in matchAndRewrite()
1708 ArrayRef<int64_t>(dims_vals), value_elem.getType().getElementType()); in matchAndRewrite()