Searched refs:GetGpuDeviceArrayOnDevice (Results 1 – 9 of 9) sorted by relevance
50 GetGpuDeviceArrayOnDevice(&ind_ptrs_data); in SparseConcatKernel()52 GetGpuDeviceArrayOnDevice(&val_ptrs_data); in SparseConcatKernel()54 GetGpuDeviceArrayOnDevice(&nnz_scan_data); in SparseConcatKernel()56 GetGpuDeviceArrayOnDevice(&concat_size_scan_data); in SparseConcatKernel()58 GetGpuDeviceArrayOnDevice(&output_shape_data); in SparseConcatKernel()92 GetGpuDeviceArrayOnDevice(&val_ptrs_data); in SparseConcatPermuteKernel()94 GetGpuDeviceArrayOnDevice(&nnz_scan_data); in SparseConcatPermuteKernel()96 GetGpuDeviceArrayOnDevice(&output_shape_data); in SparseConcatPermuteKernel()
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()
65 GetGpuDeviceArrayOnDevice(&mutable_this->input_start_data_); in operator ()()67 GetGpuDeviceArrayOnDevice(&mutable_this->input_size_data_); in operator ()()96 GetGpuDeviceArrayOnDevice(&input_start_data); in SparseSliceGatherKernel()
37 int32* data_indices = GetGpuDeviceArrayOnDevice(&input_indices); in DynamicStitchKernel()38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel()
169 GetGpuDeviceArrayOnDevice(&output_indices_data); in SparseSplitScatterKernel()171 GetGpuDeviceArrayOnDevice(&output_values_data); in SparseSplitScatterKernel()
37 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()