Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbufferize.cc49 auto elements_attr = op.value().cast<DenseElementsAttr>(); in matchAndRewrite() local
54 rewriter.create<ConstantOp>(loc, elements_attr.getValue({})); in matchAndRewrite()
62 bool all_same_elems = elements_attr.isSplat(); in matchAndRewrite()
65 value = rewriter.create<ConstantOp>(loc, elements_attr.getSplatValue()); in matchAndRewrite()
66 for (auto en : llvm::enumerate(elements_attr.getAttributeValues())) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc811 DenseElementsAttr *elements_attr) const { in IsScalarizableSplatConstant()
812 if (!matchPattern(value, m_Constant(elements_attr))) { in IsScalarizableSplatConstant()
824 return elements_attr->isSplat(); in IsScalarizableSplatConstant()
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-attribute.td108 ElementsAttr:$elements_attr,
146 // DEF: ::mlir::ElementsAttr BOp::elements_attr()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2287 if (auto elements_attr = operands[0].dyn_cast_or_null<ElementsAttr>()) { in fold() local
2288 auto rank = static_cast<int32_t>(elements_attr.getType().getRank()); in fold()
2328 auto elements_attr = operands[0].dyn_cast_or_null<DenseIntElementsAttr>(); in fold() local
2329 if (!elements_attr) { in fold()
2365 return elements_attr.mapValues(result_element_type, cast); in fold()