- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for toRType (0.08 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
OpBuilder &builder) { auto broadcast_dims = GetI64ElementsAttr({feature_dim}, &builder); auto to_type = mlir::cast<RankedTensorType>(broadcast_to.getType()); auto result_shape = builder.create<shape::ShapeOfOp>(loc, broadcast_to); auto result_extents_type = GetExtentsTensorTypeFor(to_type); auto result_extents = builder.create<shape::ToExtentTensorOp>( loc, result_extents_type, result_shape);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)