Searched refs:begin_indices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1762 SmallVector<int64_t, 4> begin_indices(input_shape.size(), 0); in matchAndRewrite() local
1770 op.input(), GetI64ElementsAttr(begin_indices, &rewriter), in matchAndRewrite()
2941 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
2950 begin_indices[dim_index] = i * slice_size; in matchAndRewrite()
2954 GetI64ElementsAttr(begin_indices, &rewriter), in matchAndRewrite()
3053 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
3062 end_indices[dim_index] = begin_indices[dim_index] + split_sizes[i]; in matchAndRewrite()
3064 op.getLoc(), op.value(), GetI64ElementsAttr(begin_indices, &rewriter), in matchAndRewrite()
3068 begin_indices[dim_index] = end_indices[dim_index]; in matchAndRewrite()
3100 ArrayRef<int64_t> begin_indices, in rewriteWithConstantBegin() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1098 DenseIntElementsAttr begin_indices; in Verify() local
1099 if (matchPattern(op.begin(), m_Constant(&begin_indices))) { in Verify()
1106 for (const APInt &raw_begin_index : begin_indices.getValues<APInt>()) { in Verify()
|
| D | tf_generated_ops.td | 15402 // `begin_indices`, `end_indices`, and `strides` with their canonical
15451 // `begin_indices`, `end_indices`, and `strides` with their canonical
|