Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc105 auto size_attr = DenseElementsAttr::get<int64_t>(vector3_type, slice_size); in sliceInput() local
107 auto size = rewriter.create<TF::ConstOp>(loc, vector3_type, size_attr); in sliceInput()