Searched defs:slice_begin (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| H A D | slice_op_test.py | 320 def _testGradientSlice(self, input_shape, slice_begin, slice_size): argument
345 def _testGradientSliceTape(self, input_shape, slice_begin, slice_size): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | strided_slice_op.cc | 290 absl::InlinedVector<int64_t, 4> slice_begin, slice_end, slice_strides; in Compile() local
712 absl::InlinedVector<xla::XlaOp, 4> slice_begin; in Compile() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | ValidateCompressedIndicesCommon.h | 126 const auto* RESTRICT slice_begin = ptr_idx_batch + cidx; in _check_idx_sorted_distinct_vals_slices_with_cidx() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | conv_spmd_expander.cc | 264 mlir::Value slice_begin = in HandleConv() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| H A D | device_transform_patterns.cc | 161 SmallVector<int32_t, 4> slice_begin; in SliceOutputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 2000 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedBoundRanges()
2169 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedShapeAndBoundRanges()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer.cc | 3815 NodeDef* slice_begin; in GetSimpleSliceAxis() local
3922 NodeDef* slice_begin; in GetStridedSliceAxis() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | graph_properties.cc | 1847 const Tensor* slice_begin = ic->input_tensor(1); in MaybeUpdateNodeContextOutput() local
|