- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetIndexAttr (0.18 sec)
-
tensorflow/compiler/mlir/tfr/passes/decompose.cc
for (int i = res.index(), j = 0; i < op->getNumResults(); i++, j++) { auto index = builder.create<mlir::arith::ConstantOp>( op->getLoc(), builder.getIndexAttr(j)); auto element_op = builder.create<GetElementOp>( op->getLoc(), unconstrainted_tensor_type, new_op.getResult(res.index()), index.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
Value lhs_shape = rewriter->create<shape::ShapeOfOp>(loc, lhs); Value rhs_shape = rewriter->create<shape::ShapeOfOp>(loc, rhs); Value const_neg2 = rewriter->create<arith::ConstantOp>(loc, rewriter->getIndexAttr(-2)); auto shape_type = shape::ShapeType::get(rewriter->getContext()); auto lhs_splitted = rewriter->create<shape::SplitAtOp>( loc, TypeRange{shape_type, shape_type}, lhs_shape, const_neg2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)