Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc71 auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); in createReshapeOp() local
73 rewriter.create<TF::ConstOp>(loc, shape_spec_type, constant_attr); in createReshapeOp()
Deinsum.cc78 auto constant_attr = DenseElementsAttr::get(shape_spec_type, shape); in createReshapeOp() local
80 rewriter->create<ConstantOp>(loc, shape_spec_type, constant_attr); in createReshapeOp()