Home
last modified time | relevance | path

Searched defs:slice_index (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_split_op_gpu.cu.cc74 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
Dsparse_split_op.cc48 for (int slice_index = 0; slice_index < num_split; ++slice_index) { in operator ()() local
/external/mesa3d/src/gallium/frontends/va/
Dpicture_h264.c182 uint32_t slice_index = context->desc.h264.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferH264() local
Dpicture_hevc.c231 uint32_t slice_index = context->desc.h265.slice_parameter.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferHEVC() local
Dpicture_av1.c408 uint32_t slice_index = context->desc.av1.slice_parameter.slice_count + buffer_idx; in vlVaHandleSliceParameterBufferAV1() local
/external/rust/android-crates-io/crates/bit_field/src/
Dlib.rs288 let slice_index = bit / T::BIT_LENGTH; in get_bit() localVariable
325 let slice_index = bit / T::BIT_LENGTH; in set_bit() localVariable
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h250 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
/external/pigweed/pw_bytes/rust/
Dpw_bytes.rs63 let mut slice_index = 0; localVariable
/external/eigen/unsupported/test/
Dcxx11_tensor_morphing.cpp311 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
/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc431 int slice_index) TF_ATTRIBUTE_NOINLINE { in SliceAndMaybePadState()
740 int slice_index) TF_ATTRIBUTE_NOINLINE { in MaybeUnpadAndAssignState()
/external/libgav1/src/
Dfilm_grain.cc573 const int slice_index = job_index / num_planes; in BlendNoiseChromaWorker() local
/external/pytorch/aten/src/ATen/native/cuda/
Dim2col.cuh269 const auto slice_index = index % n; in col2im_batched_kernel() local
/external/cronet/stable/third_party/rust/chromium_crates_io/vendor/nom-7.1.3/src/
Dtraits.rs324 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
/external/rust/android-crates-io/crates/nom/src/
Dtraits.rs324 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
/external/cronet/tot/third_party/rust/chromium_crates_io/vendor/nom-7.1.3/src/
Dtraits.rs324 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
/external/python/cpython3/Objects/mimalloc/
Dsegment.c603 static void mi_segment_span_free(mi_segment_t* segment, size_t slice_index, size_t slice_count, boo… in mi_segment_span_free()
699 static mi_page_t* mi_segment_span_allocate(mi_segment_t* segment, size_t slice_index, size_t slice_… in mi_segment_span_allocate()
/external/mesa3d/src/intel/compiler/
Dbrw_nir_lower_cooperative_matrix.c707 nir_def *slice_index = nir_udiv_imm(b, dst_index.ssa, packing_factor); in lower_cmat_instr() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc390 int slice_index = flatten_index / slice_size; in ExtractMinMaxFromAttr() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc1313 llvm_ir::IrArray::Index slice_index(multidim_index, outer_array.GetShape(), in SliceOutInnerArray() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2524 auto slice_index = SliceFirstK(index_gte, &b_, sort_dim, k.value()); in HandleSort() local