Home
last modified time | relevance | path

Searched refs:begin_i (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc274 int64& begin_i = (*begin)[i]; in ValidateStridedSliceOp() local
316 int64 x_fwd = begin_i < 0 ? dim_i + begin_i : begin_i; in ValidateStridedSliceOp()
317 begin_i = x_fwd; in ValidateStridedSliceOp()
318 end_i = begin_i + 1; in ValidateStridedSliceOp()
321 "slice index ", begin_i, " of dimension ", i, " out of bounds."); in ValidateStridedSliceOp()
324 begin_i = canonical(begin_i, 0); in ValidateStridedSliceOp()
329 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()
340 interval_length = end_i - begin_i; in ValidateStridedSliceOp()
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass.cc206 Output begin_i = ops::Slice( in ComputeSliceSize() local
215 input_shape_i, begin_i)); in ComputeSliceSize()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1775 int64_t begin_i = begin[i]; in CalculateSlicedShapeFromDenseIndices() local
1795 begin_i = canonicalize(begin_i, 0); in CalculateSlicedShapeFromDenseIndices()
1798 int64_t interval_len = end_i - begin_i; in CalculateSlicedShapeFromDenseIndices()
1806 begin[i] = begin_i; in CalculateSlicedShapeFromDenseIndices()
1810 end[i] = begin_i + 1; in CalculateSlicedShapeFromDenseIndices()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1557 for (auto begin_i : llvm::enumerate(begin)) { in Verify() local
1558 if (begin_i.value().getSExtValue() < 0) { in Verify()
1580 int begin_i = in Verify() local
1585 if (begin_i > dim_i) { in Verify()
1589 if (size_i >= 0 && begin_i + size_i > dim_i) { in Verify()