Searched refs:GetGpuDeviceArrayOnDevice (Results 1 – 5 of 5) sorted by relevance
37 int32* data_indices = GetGpuDeviceArrayOnDevice(&input_indices); in DynamicStitchKernel()38 const T** data_ptrs = GetGpuDeviceArrayOnDevice(&input_ptrs); in DynamicStitchKernel()
82 T** output_ptrs = GetGpuDeviceArrayOnDevice(&output_ptr_data); in SplitOpKernel()120 T** output_ptrs = GetGpuDeviceArrayOnDevice(&output_ptr_data); in split_v_kernel()121 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in split_v_kernel()177 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()
37 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE ValueType* GetGpuDeviceArrayOnDevice( in GetGpuDeviceArrayOnDevice() function
40 const float* boundaries = GetGpuDeviceArrayOnDevice(&boundaries_array); in BucketizeCustomKernel()