Home
last modified time | relevance | path

Searched defs:linearIndex (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/utils/
H A Dcodegen_utils.cc51 Value linearIndex, ArrayRef<Value> shape) { in calcMultiDimIndex()
85 Value linearIndex, Value memref) { in calcMultiDimIndex()
103 Value linearIndex, in calcMultiDimIndexForFirstOperand()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DIndexing.cu365 Tensor linearIndex; in computeLinearIndex() local
437 Tensor linearIndex, src, expandedValue = value; in index_put_with_sort_kernel() local
570 Tensor linearIndex, src, expandedValue = value; in index_put_with_sort_quantized() local
745 for (IndexType linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in indexFuncSmallIndex() local
784 for (IndexType linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in indexFuncLargeIndex() local
1235 for (IndexType linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in indexSelectSmallIndex() local
1269 for (IndexType linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in indexSelectLargeIndex() local
H A DSortUtils.cuh100 const IndexType linearIndex = blockIndex * blockDim.y + threadIdx.y; in C10_LAUNCH_BOUNDS_1() local
177 const IndexType linearIndex = blockIndex * blockDim.y + threadIdx.y; in C10_LAUNCH_BOUNDS_1() local
H A DRNN.cu112 for (index_type linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in lstm_cell_forward() local
199 for (index_type linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in lstm_cell_backward() local
263 for (index_type linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in gru_cell_forward() local
330 for (index_type linearIndex = blockIdx.x * blockDim.x + threadIdx.x; in gru_cell_backward() local
H A DDropout.cu75 for (IndexType linearIndex = idx * VEC; in fused_dropout_kernel_vec() local
152 for (IndexType linearIndex = idx; in fused_dropout_kernel() local
H A DLegacyThrustHelpers.cu22 void index_put_with_sort_kernel_thrust_helper(Tensor &linearIndex, Tensor &orig_indices, Tensor &so… in index_put_with_sort_kernel_thrust_helper()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDAApplyUtils.cuh277 for (IndexType linearIndex = (blockIdx.x * blockDim.x + threadIdx.x) * step; in C10_LAUNCH_BOUNDS_2() local
366 for (IndexType linearIndex = (blockIdx.x * blockDim.x + threadIdx.x) * step; in C10_LAUNCH_BOUNDS_2() local
/aosp_15_r20/external/deqp/external/vulkancts/modules/vulkan/sparse_resources/
H A DvktSparseResourcesImageSparseResidency.cpp475 const uint32_t linearIndex = in iterate() local
922 const uint32_t linearIndex = in iterate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc826 llvm::Value* linearIndex = in EmitPadToStatic() local
948 llvm::Value* linearIndex = in EmitSliceToDynamic() local