Searched refs:GetGpuDeviceArrayOnDevice (Results 1 – 6 of 6) sorted by relevance
37 int32* data_indices = GetGpuDeviceArrayOnDevice(&input_indices); in DynamicStitchKernel()38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel()
78 T** output_ptrs = GetGpuDeviceArrayOnDevice(&output_ptr_data); in SplitOpKernel()116 T** output_ptrs = GetGpuDeviceArrayOnDevice(&output_ptr_data); in split_v_kernel()117 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in split_v_kernel()174 T** output_ptrs = GetGpuDeviceArrayOnDevice(&output_ptr_data); in SplitVOpKernel_fixed()
40 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_fixed_kernel()65 const T** input_ptrs = GetGpuDeviceArrayOnDevice(&input_ptr_data); in concat_variable_kernel()66 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in concat_variable_kernel()
38 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE ValueType* GetGpuDeviceArrayOnDevice( in GetGpuDeviceArrayOnDevice() function
39 const float* boundaries = GetGpuDeviceArrayOnDevice(&boundaries_array); in BucketizeCustomKernel()
198 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in COOMatrixToSparseTensorKernel3D()265 const int* batch_ptr = GetGpuDeviceArrayOnDevice(&batch_ptr_s); in CSRSparseMatrixBatchMulVecKernel3D()386 const int* cuda_ptr = GetGpuDeviceArrayOnDevice(&cuda_ptr_s); in CopyFromGpuDeviceArrayToLocal()