| /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 | 721 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/ |
| D | resize_nearest_neighbor.h | 75 const int batch_offset = input_shape.Dims(1) * row_offset; in ResizeNearestNeighbor() local
94 input_ptr += batch_offset; in ResizeNearestNeighbor()
|
| D | batch_matmul.h | 180 const float batch_offset = static_cast<float>(ioff_ptr2[j]); in BatchMatMul() local
188 total -= row_sum * batch_offset; in BatchMatMul()
|
| D | portable_tensor_utils.cc | 183 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/ |
| 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/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 231 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()
|
| D | sse_tensor_utils.cc | 104 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()
|
| D | optimized_ops.h | 6064 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/ |
| 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/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 410 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()
|
| 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/python/data/experimental/kernel_tests/ |
| D | bucket_by_sequence_length_test.py | 428 for batch_offset in batch_offsets:
429 batch_start_len = bucket_start_len + batch_offset
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 601 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)
|