| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_split_op_gpu.cu.cc | 74 inline Index GetSliceShape(const Index slice_index, const Index split_size, in GetSliceShape()
139 for (int slice_index : GpuGridRangeX<int>(num_split)) { in SparseSplitFindSliceEndsKernel() local
175 int slice_index = in SparseSplitScatterKernel() local
351 for (int slice_index = 0; slice_index < num_split; ++slice_index) { in AllocateOutputs() local
|
| H A D | sparse_split_op.cc | 48 for (int slice_index = 0; slice_index < num_split; ++slice_index) { in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/ |
| H A D | sparse_tensor.h | 250 static inline int GetSliceShape(const int slice_index, const int split_size, in GetSliceShape()
530 int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
554 const int slice_index = GetSliceIndex(dim, split_size, residual); in Split() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/bit_field/src/ |
| D | lib.rs | 288 let slice_index = bit / T::BIT_LENGTH; in get_bit() localVariable
325 let slice_index = bit / T::BIT_LENGTH; in set_bit() localVariable
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/va/ |
| H A D | picture_h264.c | 182 uint32_t slice_index = context->desc.h264.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferH264() local
|
| H A D | picture_hevc.c | 231 uint32_t slice_index = context->desc.h265.slice_parameter.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferHEVC() local
|
| H A D | picture_av1.c | 404 uint32_t slice_index = context->desc.av1.slice_parameter.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferAV1() local
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_morphing.cpp | 311 int slice_index = i + 3 * (j + 5 * (k + 7 * l)); in test_slice_raw_data() local
326 int slice_index = l + 11 * (k + 7 * (j + 5 * i)); in test_slice_raw_data() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops.cc | 431 int slice_index) TF_ATTRIBUTE_NOINLINE { in SliceAndMaybePadState()
740 int slice_index) TF_ATTRIBUTE_NOINLINE { in MaybeUnpadAndAssignState()
|
| /aosp_15_r20/external/pigweed/pw_bytes/rust/ |
| H A D | pw_bytes.rs | 63 let mut slice_index = 0; localVariable
|
| /aosp_15_r20/external/rust/android-crates-io/crates/nom/src/ |
| D | traits.rs | 324 fn slice_index(&self, count: usize) -> Result<usize, Needed>; in slice_index() method
356 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
401 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/nom-7.1.3/src/ |
| H A D | traits.rs | 324 fn slice_index(&self, count: usize) -> Result<usize, Needed>; in slice_index() method
356 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
401 fn slice_index(&self, count: usize) -> Result<usize, Needed> { in slice_index() method
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | im2col.cuh | 269 const auto slice_index = index % n; in col2im_batched_kernel() local
|
| /aosp_15_r20/external/libgav1/src/ |
| H A D | film_grain.cc | 564 const int slice_index = job_index / num_planes; in BlendNoiseChromaWorker() local
|
| /aosp_15_r20/external/mesa3d/src/intel/compiler/ |
| H A D | brw_nir_lower_cooperative_matrix.c | 707 nir_def *slice_index = nir_udiv_imm(b, dst_index.ssa, packing_factor); in lower_cmat_instr() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | quantization_utils.cc | 390 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr() local
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/alloc/tests/ |
| H A D | str.rs | 328 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0/lib/rustlib/src/rust/library/alloc/tests/ |
| H A D | str.rs | 328 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/alloc/tests/ |
| H A D | str.rs | 328 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.80.1/lib/rustlib/src/rust/library/alloc/tests/ |
| H A D | str.rs | 328 mod slice_index { module
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | dot_op_emitter.cc | 1313 llvm_ir::IrArray::Index slice_index(multidim_index, outer_array.GetShape(), in SliceOutInnerArray() local
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/alloc/tests/ |
| H A D | str.rs | 328 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-musl-x86/1.81.0/lib/rustlib/src/rust/library/core/tests/ |
| H A D | slice.rs | 1415 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.81.0.u1/lib/rustlib/src/rust/library/core/tests/ |
| H A D | slice.rs | 1415 mod slice_index { module
|
| /aosp_15_r20/prebuilts/rust/linux-x86/1.80.1/lib/rustlib/src/rust/library/core/tests/ |
| H A D | slice.rs | 1415 mod slice_index { module
|