Searched defs:batch_ptr (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | kernels_test.cc | 32 Tensor batch_ptr(DT_INT32, {2}); in TEST() local
55 Tensor batch_ptr(DT_INT32, {4}); in TEST() local
|
| H A D | sparse_tensor_to_csr_sparse_matrix_op.cc | 89 Tensor batch_ptr(cpu_allocator(), DT_INT32, batch_ptr_shape); in Compute() local
207 auto batch_ptr = batch_ptr_t.vec<int32>(); in ComputeAsync() local
|
| H A D | dense_to_csr_sparse_matrix_op.cc | 103 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
259 auto batch_ptr = batch_ptr_t.vec<int32>(); in ComputeAsync() local
|
| H A D | kernels.cc | 32 TTypes<int64_t>::ConstMatrix indices, TTypes<int32>::Vec batch_ptr, in operator ()()
|
| H A D | kernels_gpu.cu.cc | 198 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in COOMatrixToSparseTensorKernel3D() local
265 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in CSRSparseMatrixBatchMulVecKernel3D() local
|
| H A D | sparse_cholesky_op.cc | 91 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
|
| H A D | sparse_mat_mul_op.cc | 161 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
|
| /aosp_15_r20/external/libdrm/intel/ |
| H A D | test_decode.c | 73 void *batch_ptr; in dump_batch() local
90 void *ref_ptr, *batch_ptr; in compare_batch() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/crocus/ |
| H A D | crocus_state.c | 7441 void *batch_ptr; local
|