| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuConcatenateBatchKernel.cpp | 47 void batch_concat(const ITensor *src, ITensor *dst, unsigned int batch_offset, const Window &window) in batch_concat()
132 Status validate_arguments(const ITensorInfo *src, unsigned int batch_offset, const ITensorInfo *dst) in validate_arguments()
149 void CpuConcatenateBatchKernel::configure(const ITensorInfo *src, unsigned int batch_offset, ITenso… in configure()
185 unsigned int batch_offset, in validate()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDASparseDescriptors.cpp | 59 cusparseDnMatDescr_t createRawDnMatDescriptor(const Tensor& input, int64_t batch_offset, bool is_co… in createRawDnMatDescriptor()
115 CuSparseDnMatDescriptor::CuSparseDnMatDescriptor(const Tensor& input, int64_t batch_offset) { in CuSparseDnMatDescriptor()
119 …arseConstDnMatDescriptor::CuSparseConstDnMatDescriptor(const Tensor& input, int64_t batch_offset) { in CuSparseConstDnMatDescriptor()
141 CuSparseSpMatCsrDescriptor::CuSparseSpMatCsrDescriptor(const Tensor& input, int64_t batch_offset) { in CuSparseSpMatCsrDescriptor()
|
| /aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/ |
| H A D | ClBatchConcatenateKernel.cpp | 45 Status validate_arguments(const ITensorInfo *src, unsigned int batch_offset, const ITensorInfo *dst) in validate_arguments()
67 …gure(const CLCompileContext &compile_context, ITensorInfo *src, unsigned int batch_offset, ITensor… in configure()
125 unsigned int batch_offset, in validate()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | resize_nearest_neighbor.h | 76 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
|
| H A D | batch_matmul.h | 181 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
|
| H A D | portable_tensor_utils.cc | 183 const int32_t batch_offset = input_offset[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
|
| /aosp_15_r20/external/mesa3d/src/intel/tools/ |
| H A D | aub_write.c | 714 uint64_t batch_offset) in aub_dump_ring_buffer_execlist()
765 uint64_t batch_offset, in aub_dump_ring_buffer_legacy()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gather_functor_batched.h | 68 SliceIndex batch_offset = batch_idx * indices_size; in HandleCopiesBatched() local
|
| H A D | resource_variable_ops.cc | 792 int64_t const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets() local
|
| /aosp_15_r20/external/igt-gpu-tools/tools/ |
| H A D | aubdump.c | 716 aub_dump_execlist(uint64_t batch_offset, int ring_flag) in aub_dump_execlist()
794 aub_dump_ringbuffer(uint64_t batch_offset, uint64_t offset, int ring_flag) in aub_dump_ringbuffer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | sparse_matrix.h | 199 inline int batch_offset(int batch) const { in batch_offset() function
|
| H A D | kernels_gpu.cu.cc | 410 const int batch_offset = local_batch_ptr[batch]; in CSRSparseMatrixSoftmaxKernel3D() local
|
| /aosp_15_r20/external/mesa3d/src/intel/executor/ |
| H A D | executor_main.c | 683 const int batch_offset = ec.batch_start - ec.bo.batch.addr; in l_execute() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/crocus/ |
| H A D | crocus_batch.c | 432 crocus_command_reloc(struct crocus_batch *batch, uint32_t batch_offset, in crocus_command_reloc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | batch_matmul.h | 231 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
|
| H A D | sse_tensor_utils.cc | 207 const int32_t batch_offset = input_offset ? input_offset[batch] : 0; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local
|
| H A D | optimized_ops.h | 5576 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
|
| /aosp_15_r20/external/mesa3d/src/intel/common/tests/ |
| H A D | mi_builder_test.cpp | 145 uint32_t batch_offset; member in mi_builder_test
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | operator-run.c | 720 size_t batch_offset, in compute_q8add_strided()
|