Searched refs:begin_i (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 278 int64_t& begin_i = (*begin)[i]; in ValidateStridedSliceOp() local
321 int64_t x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i; in ValidateStridedSliceOp()
322 begin_i = x_fwd; in ValidateStridedSliceOp()
323 end_i = begin_i + 1; in ValidateStridedSliceOp()
326 "slice index ", begin_i, " of dimension ", i, " out of bounds."); in ValidateStridedSliceOp()
329 begin_i = canonical(begin_i, 0); in ValidateStridedSliceOp()
334 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()
345 interval_length = end_i - begin_i; in ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 206 Output begin_i = ops::Slice( in ComputeSliceSize() local
215 input_shape_i, begin_i)); in ComputeSliceSize()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1915 int64_t begin_i = begin[i]; in CalculateSlicedShapeFromDenseIndices() local
1935 begin_i = canonicalize(begin_i, 0); in CalculateSlicedShapeFromDenseIndices()
1938 int64_t interval_len = end_i - begin_i; in CalculateSlicedShapeFromDenseIndices()
1946 begin[i] = begin_i; in CalculateSlicedShapeFromDenseIndices()
1950 end[i] = begin_i + 1; in CalculateSlicedShapeFromDenseIndices()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2029 for (const auto &begin_i : llvm::enumerate(begin)) { in verify() local
2030 if (begin_i.value().getSExtValue() < 0) { in verify()
2052 int begin_i = begin.getValues<APInt>()[i].getSExtValue(); in verify() local
2055 if (begin_i > dim_i) { in verify()
2059 if (size_i >= 0 && begin_i + size_i > dim_i) { in verify()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1593 auto create_slice_op = [&](int32_t axis_i, int32_t begin_i, int32_t size_i, in matchAndRewrite()
1596 begin_values[axis_i] = begin_i; in matchAndRewrite()
|