Searched refs:getIndexAttr (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tensorflow_abi_knowledge_propagation.cc | 111 b.getIndexAttr( in runOnFunction()
119 b.getIndexAttr(0)); in runOnFunction()
132 b.getIndexAttr(const_stride->second)); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| D | ShapeToShapeLowering.cpp | 38 ->materializeConstant(rewriter, rewriter.getIndexAttr(1), in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/IR/ |
| D | Shape.cpp | 500 build(builder, result, builder.getIndexAttr(value)); in build()
658 auto dimAttr = builder.getIndexAttr(dim); in build()
679 return builder.getIndexAttr(rank); in fold()
743 return builder.getIndexAttr(product.getLimitedValue()); in fold()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 98 IntegerAttr Builder::getIndexAttr(int64_t value) { in getIndexAttr() function in Builder
280 return getIndexAttr(0); in getZeroAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | decompose.cc | 212 builder.create<ConstantOp>(op->getLoc(), builder.getIndexAttr(j)); in RewriteUnregisteredTFOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 928 global_var->setAttr("lmhlo.alloc", builder_.getIndexAttr(slice.index())); in EmitConstant()
1498 arg_attr_list.set("lmhlo.alloc", builder_.getIndexAttr(alloc->index())); in Initialize()
1500 builder_.getIndexAttr(alloc->parameter_number())); in Initialize()
1507 arg_attr_list.set("lmhlo.alloc", builder_.getIndexAttr(alloc->index())); in Initialize()
|
| D | legalize_tf.cc | 2777 rewriter->create<ConstantOp>(loc, rewriter->getIndexAttr(-2)); in BroadcastBatchMatMulV2Operands()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 117 IntegerAttr getIndexAttr(int64_t value);
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.cc | 489 rewriter.getIndexAttr(num_tensors)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 1470 Value bnd = rewriter.create<ConstantOp>(loc, rewriter.getIndexAttr(d)); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/IR/ |
| D | Ops.cpp | 1386 return builder.getIndexAttr(shapedTy.getShape()[index.getInt()]); in fold()
|