Searched refs:GetI64ElementsAttrForSeq (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 162 static DenseIntElementsAttr GetI64ElementsAttrForSeq(int start, int end, in GetI64ElementsAttrForSeq() function
436 GetI64ElementsAttrForSeq(0, result_type.getRank(), &builder); in StaticBinaryBroadcast()
487 auto prefix_dims = GetI64ElementsAttrForSeq(0, lower_rank, &builder); in CommonPrefixBroadcast()
533 auto broadcast_dims = GetI64ElementsAttrForSeq(0, rank, &builder); in BroadcastToShapeOf()
1110 GetI64ElementsAttrForSeq(0, num_spatial_dims, builder); in GetConvDimensionNumbersAttr()
1650 /*offset_dims=*/GetI64ElementsAttrForSeq(0, num_dims - 2, &rewriter), in matchAndRewrite()
3940 GetI64ElementsAttrForSeq(updates_rank - window_dims, updates_rank, in matchAndRewrite()
3942 GetI64ElementsAttrForSeq(0, num_index_dims, &rewriter), in matchAndRewrite()
3943 GetI64ElementsAttrForSeq(0, num_index_dims, &rewriter), in matchAndRewrite()
4224 GetI64ElementsAttrForSeq(0, num_spatial_dims, &rewriter); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.td | 203 "GetI64ElementsAttrForSeq("
|
| D | lower_tf.cc | 51 static DenseIntElementsAttr GetI64ElementsAttrForSeq(int start, int end, in GetI64ElementsAttrForSeq() function
|