Home
last modified time | relevance | path

Searched refs:batch_offset (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_functor_batched.h68 SliceIndex batch_offset = batch_idx * indices_size; in HandleCopiesBatched() local
73 SliceIndex b_offset_next = batch_offset; in HandleCopiesBatched()
89 indices(batch_offset + indices_idx)); in HandleCopiesBatched()
92 result = batch_offset + indices_idx; in HandleCopiesBatched()
118 batch_offset = b_offset_next; in HandleCopiesBatched()
Dresource_variable_ops.cc721 int64 const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets() local
725 indices_flat(dest_idx++) += batch_offset * batch_idx; in AddBatchOffsets()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dresize_nearest_neighbor.h75 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
94 input_ptr += batch_offset; in ResizeNearestNeighbor()
Dbatch_matmul.h180 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
188 total -= row_sum * batch_offset; in BatchMatMul()
Dportable_tensor_utils.cc183 const int32_t batch_offset = input_offset[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
199 dotprod -= row_sums[row] * batch_offset; in PortableMatrixBatchVectorMultiplyAccumulate()
/external/mesa3d/src/intel/common/tests/
Dgen_mi_builder_test.cpp131 uint32_t batch_offset; member in gen_mi_builder_test
222 batch_offset = 0; in SetUp()
258 void *ptr = (void *)((char *)batch_map + batch_offset); in emit_dwords()
259 batch_offset += num_dwords * 4; in emit_dwords()
260 assert(batch_offset < BATCH_BO_SIZE); in emit_dwords()
270 if (batch_offset & 4) in submit_batch()
296 execbuf.batch_len = batch_offset; in submit_batch()
/external/mesa3d/src/mesa/drivers/dri/i965/
Dintel_batchbuffer.h67 uint32_t batch_offset,
72 uint32_t batch_offset,
Dintel_batchbuffer.c1010 brw_batch_reloc(struct intel_batchbuffer *batch, uint32_t batch_offset, in brw_batch_reloc() argument
1014 assert(batch_offset <= batch->batch.bo->size - sizeof(uint32_t)); in brw_batch_reloc()
1016 return emit_reloc(batch, &batch->batch_relocs, batch_offset, in brw_batch_reloc()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h231 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
236 const int32x4_t input_offset0 = vdupq_n_s32(-batch_offset); in BatchMatMul()
237 const int32x4_t input_offset1 = vdupq_n_s32(-batch_offset); in BatchMatMul()
262 accum_scratch[idx] -= woff_ptr2[i] * batch_offset; in BatchMatMul()
Dsse_tensor_utils.cc104 const int32_t batch_offset = input_offset ? input_offset[batch] : 0; in SseMatrixBatchVectorMultiplyAccumulateImpl() local
113 row_sums && batch_offset ? batch_offset * row_sums[row] : 0; in SseMatrixBatchVectorMultiplyAccumulateImpl()
Doptimized_ops.h6064 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
6092 input_ptr += batch_offset; in ResizeNearestNeighbor()
/external/mesa3d/src/intel/tools/
Daub_write.c723 uint64_t batch_offset) in aub_dump_ring_buffer_execlist() argument
729 dword_out(aub, batch_offset & 0xFFFFFFFF); in aub_dump_ring_buffer_execlist()
730 dword_out(aub, batch_offset >> 32); in aub_dump_ring_buffer_execlist()
774 uint64_t batch_offset, in aub_dump_ring_buffer_legacy() argument
793 aub_write_reloc(&aub->devinfo, &ringbuffer[ring_count + 1], 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() argument
756 dword_out(batch_offset & 0xFFFFFFFF); in aub_dump_execlist()
757 dword_out(batch_offset >> 32); in aub_dump_execlist()
794 aub_dump_ringbuffer(uint64_t batch_offset, uint64_t offset, int ring_flag) in aub_dump_ringbuffer() argument
811 write_reloc(&ringbuffer[ring_count + 1], batch_offset); in aub_dump_ringbuffer()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dkernels_gpu.cu.cc410 const int batch_offset = local_batch_ptr[batch]; in CSRSparseMatrixSoftmaxKernel3D() local
412 CalculateRowSoftmax(batch_offset + ldg(row_ptr + row_offset), in CSRSparseMatrixSoftmaxKernel3D()
413 batch_offset + ldg(row_ptr + row_offset + 1), logits, in CSRSparseMatrixSoftmaxKernel3D()
Dcsr_sparse_matrix_to_sparse_tensor_op.cc187 &coo_row_ind(csr_sparse_matrix->batch_offset(i)), nnz_i); in Compute()
Dcsr_sparse_matrix_to_dense_op.cc181 &coo_row_ind(csr_sparse_matrix->batch_offset(i)), nnz_i); in Compute()
Dsparse_matrix.h199 inline int batch_offset(int batch) const { in batch_offset() function
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dbucket_by_sequence_length_test.py428 for batch_offset in batch_offsets:
429 batch_start_len = bucket_start_len + batch_offset
/external/tensorflow/tensorflow/python/ops/
Dctc_ops.py601 batch_offset = math_ops.range(batch_size, dtype=unique_y.dtype) * num_labels
602 indices = unique_y + array_ops.expand_dims(batch_offset, axis=-1)