| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_functor_batched.h | 68 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()
|
| D | resource_variable_ops.cc | 735 int64 const batch_offset = params.dim_size(batch_dims_); in AddBatchOffsets() local
739 indices_flat(dest_idx++) += batch_offset * batch_idx; in AddBatchOffsets()
|
| /external/mesa3d/src/intel/common/tests/ |
| D | gen_mi_builder_test.cpp | 131 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/ |
| D | intel_batchbuffer.h | 67 uint32_t batch_offset,
72 uint32_t batch_offset,
|
| D | intel_batchbuffer.c | 1010 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/mesa3d/src/intel/tools/ |
| D | aub_write.c | 723 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/ |
| D | aubdump.c | 716 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/python/data/experimental/kernel_tests/ |
| D | bucket_by_sequence_length_test.py | 424 for batch_offset in batch_offsets:
425 batch_start_len = bucket_start_len + batch_offset
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 423 const int batch_offset = local_batch_ptr[batch]; in CSRSparseMatrixSoftmaxKernel3D() local
425 CalculateRowSoftmax(batch_offset + ldg(row_ptr + row_offset), in CSRSparseMatrixSoftmaxKernel3D()
426 batch_offset + ldg(row_ptr + row_offset + 1), logits, in CSRSparseMatrixSoftmaxKernel3D()
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 187 &coo_row_ind(csr_sparse_matrix->batch_offset(i)), nnz_i); in Compute()
|
| D | csr_sparse_matrix_to_dense_op.cc | 181 &coo_row_ind(csr_sparse_matrix->batch_offset(i)), nnz_i); in Compute()
|
| D | sparse_matrix.h | 199 inline int batch_offset(int batch) const { in batch_offset() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | portable_tensor_utils.cc | 171 const float batch_offset = input_offset[batch]; in PortableMatrixBatchVectorMultiplyAccumulate() local
181 dotprod += (*row_ptr) * (vectors[col] - batch_offset); in PortableMatrixBatchVectorMultiplyAccumulate()
|
| D | reference_ops.h | 2782 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
2799 input_ptr += batch_offset; in ResizeNearestNeighbor()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 596 batch_offset = math_ops.range(batch_size, dtype=unique_y.dtype) * num_labels
597 indices = unique_y + array_ops.expand_dims(batch_offset, axis=-1)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.cc | 1307 const int32_t batch_offset = input_offset[batch]; in NeonMatrixBatchVectorMultiplyAccumulate() local
1308 dotprod -= row_sum * batch_offset; in NeonMatrixBatchVectorMultiplyAccumulate()
|
| D | optimized_ops.h | 5856 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
5884 input_ptr += batch_offset; in ResizeNearestNeighbor()
|