Searched defs:batch_ptr (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_test.cc | 32 Tensor batch_ptr(DT_INT32, {2}); in TEST() local
55 Tensor batch_ptr(DT_INT32, {4}); in TEST() local
|
| D | kernels.cc | 31 TTypes<int64>::ConstMatrix indices, TTypes<int32>::Vec batch_ptr, in operator ()()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 79 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
191 auto batch_ptr = batch_ptr_t.vec<int32>(); in ComputeAsync() local
|
| D | dense_to_csr_sparse_matrix_op.cc | 103 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
258 auto batch_ptr = batch_ptr_t.vec<int32>(); in ComputeAsync() local
|
| D | sparse_cholesky_op.cc | 89 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
|
| 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
|
| D | sparse_mat_mul_op.cc | 161 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
|
| /external/libdrm/intel/ |
| D | test_decode.c | 73 void *batch_ptr; in dump_batch() local
89 void *ptr, *ref_ptr, *batch_ptr; in compare_batch() local
|