Searched defs:begin_indices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | slice_op.cc | 146 absl::InlinedVector<xla::XlaOp, 4> begin_indices; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2369 SmallVector<int64_t, 4> begin_indices(input_shape.size(), 0); in matchAndRewrite() local
3294 Value begin_indices = op.begin(); in matchAndRewrite() local
3549 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
3615 SmallVector<Value, 4> begin_indices(input_rank, zero); in matchAndRewrite() local
3737 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
3784 ArrayRef<int64_t> begin_indices, in rewriteWithConstantBegin()
4002 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
4034 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
5724 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
5769 SmallVector<int64_t, 4> begin_indices(value_rank, 0); in matchAndRewrite() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1224 DenseIntElementsAttr begin_indices; in Verify() local
|