Searched defs:batch_ptr (Results 1 – 10 of 10) 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 | 78 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
190 auto batch_ptr = batch_ptr_t.vec<int32>(); in ComputeAsync() local
|
| D | dense_to_csr_sparse_matrix_op.cc | 102 Tensor batch_ptr(cpu_allocator(), DT_INT32, TensorShape({batch_size + 1})); in Compute() local
257 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 | 211 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in COOMatrixToSparseTensorKernel3D() local
278 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
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | svdf.cc | 131 float* batch_ptr = state_ptr_batch; in ApplyTimeWeightsBiasAndActivation() local
331 int16_t* batch_ptr = state_ptr_batch; in EvalIntegerSVDF() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa-hifimini/ |
| D | svdf.cc | 250 int16_t* batch_ptr = state_ptr_batch; in EvalIntegerSVDF() local
|