Searched defs:begin_indices (Results 1 – 4 of 4) 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 | 2161 SmallVector<int64_t, 4> begin_indices(input_shape.size(), 0); in matchAndRewrite() local
3081 Value begin_indices = op.begin(); in matchAndRewrite() local
3346 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
3412 SmallVector<Value, 4> begin_indices(input_rank, zero); in matchAndRewrite() local
3545 SmallVector<int64_t, 4> begin_indices(input_rank, 0); in matchAndRewrite() local
3592 ArrayRef<int64_t> begin_indices, in rewriteWithConstantBegin()
3807 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
3839 SmallVector<int64_t, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
5515 SmallVector<int64_t, 4> begin_indices(value_rank, 0); in matchAndRewrite() local
5565 SmallVector<Value, 4> begin_indices, end_indices, strides; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1275 DenseIntElementsAttr begin_indices; in verify() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 1350 std::vector<int> begin_indices(input_rank); in TransformSplitVIntoSupportedOps() local
|