Home
last modified time | relevance | path

Searched defs:size_i (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dincrease_dynamism_for_auto_jit_pass.cc338 [](int64_t size_i) { return size_i >= -1; }); in ShouldRewriteSlice()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dstrided_slice_op.cc366 int64_t size_i; in ValidateStridedSliceOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc2041 for (const auto &size_i : llvm::enumerate(size)) { in verify() local
2053 int size_i = size.getValues<APInt>()[i].getSExtValue(); in verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc1594 Value input) { in matchAndRewrite()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp3119 const auto& size_i = sizes[i]; in _min_storage_size() local
3151 const auto& size_i = sym_sizes[i]; in as_strided_backward() local
3182 const auto& size_i = inp_sizes[i]; in as_strided_backward() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc1939 int64_t size_i = 0; in CalculateSlicedShapeFromDenseIndices() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc4683 int64_t size_i = (new_slice_limits[i] - new_slice_starts[i] + in HandleSlice() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1543 auto size_i = xtensor.sizes()[i]; in repeat() local