Searched defs:matrix_ptr (Results 1 – 5 of 5) 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
656 LLVMValueRef matrix_ptr = LLVMBuildAdd(builder, functions_ptr, matrix_offset, ""); in compile_jit_sample_function() 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