Home
last modified time | relevance | path

Searched defs:batch_offset (Results 1 – 18 of 18) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/
DCpuConcatenateBatchKernel.cpp47 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()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClBatchConcatenateKernel.cpp45 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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dresize_nearest_neighbor.h76 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
Dbatch_matmul.h181 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
Dportable_tensor_utils.cc183 const int32_t batch_offset = input_offset[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
/external/tensorflow/tensorflow/core/kernels/
Dgather_functor_batched.h68 SliceIndex batch_offset = batch_idx * indices_size; in HandleCopiesBatched() local
Dresource_variable_ops.cc792 int64_t const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets() local
/external/mesa3d/src/intel/tools/
Daub_write.c723 uint64_t batch_offset) in aub_dump_ring_buffer_execlist()
774 uint64_t batch_offset, in aub_dump_ring_buffer_legacy()
/external/igt-gpu-tools/tools/
Daubdump.c716 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()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_matrix.h199 inline int batch_offset(int batch) const { in batch_offset() function
Dkernels_gpu.cu.cc410 const int batch_offset = local_batch_ptr[batch]; in CSRSparseMatrixSoftmaxKernel3D() local
/external/mesa3d/src/intel/common/tests/
Dgen_mi_builder_test.cpp131 uint32_t batch_offset; member in gen_mi_builder_test
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h231 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
Dsse_tensor_utils.cc207 const int32_t batch_offset = input_offset ? input_offset[batch] : 0; in Avx2MatrixBatchVectorMultiplyAccumulateImpl() local
Doptimized_ops.h5576 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
/external/mesa3d/src/mesa/drivers/dri/i965/
Dintel_batchbuffer.c1010 brw_batch_reloc(struct intel_batchbuffer *batch, uint32_t batch_offset, in brw_batch_reloc()
/external/bcc/libbpf-tools/x86/
Dvmlinux_505.h78010 u32 batch_offset; member
Dvmlinux.h78010 u32 batch_offset; member