/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)
|