Searched defs:matrix_ptr (Results 1 – 4 of 4) sorted by relevance
33 __global__ void MatrixDiagPartV3Kernel(const T *matrix_ptr, const T *padding_value_ptr, T *diagnal_… in MatrixDiagPartV3Kernel()55 cudaError_t MatrixDiagPartV3(const T *matrix_ptr, const T *padding_value_ptr, T *diagnal_ptr, const… in MatrixDiagPartV3()
66 auto matrix_ptr = reinterpret_cast<T *>(inputs[kIndex0]->device_ptr()); in LaunchKernel() local
72 auto matrix_ptr = GetDeviceAddress<T>(inputs, kIndex0); in LaunchKernel() local
3529 double* matrix_ptr = (double*)(m_bo_1_data + n_matrix * m_bo_1_batch_size + m_bo_1_offset_matrix); in initBuffers() local3576 unsigned char* matrix_ptr = m_bo_2_data + n_matrix * m_bo_2_batch_size + m_bo_2_offset_matrix; in initBuffers() local