Searched defs:batch_offset (Results 1 – 14 of 14) sorted by relevance
75 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
180 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
183 const int32_t batch_offset = input_offset[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
68 SliceIndex batch_offset = batch_idx * indices_size; in HandleCopiesBatched() local
721 int64 const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets() local
723 uint64_t batch_offset) in aub_dump_ring_buffer_execlist()774 uint64_t batch_offset, in aub_dump_ring_buffer_legacy()
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()
199 inline int batch_offset(int batch) const { in batch_offset() function
410 const int batch_offset = local_batch_ptr[batch]; in CSRSparseMatrixSoftmaxKernel3D() local
231 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
104 const int32_t batch_offset = input_offset ? input_offset[batch] : 0; in SseMatrixBatchVectorMultiplyAccumulateImpl() local
6064 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
131 uint32_t batch_offset; member in gen_mi_builder_test
1010 brw_batch_reloc(struct intel_batchbuffer *batch, uint32_t batch_offset, in brw_batch_reloc()