Searched refs:transpose_tensor (Results 1 – 1 of 1) sorted by relevance
58 …ensor> inline prepare_matrix_for_cublas(const Tensor& tensor, bool& transpose_tensor, bool transpo… in prepare_matrix_for_cublas() argument60 transpose_tensor = tensor.is_contiguous(); in prepare_matrix_for_cublas()61 … return resolve_conj_if_indicated(tensor, transpose_result ? transpose_tensor : !transpose_tensor); in prepare_matrix_for_cublas()66 transpose_tensor = false; in prepare_matrix_for_cublas()69 transpose_tensor = true; in prepare_matrix_for_cublas()72 transpose_tensor = true; in prepare_matrix_for_cublas()77 …MaybeOwned<Tensor> inline prepare_matrix_for_cublas(const Tensor& tensor, bool& transpose_tensor) { in prepare_matrix_for_cublas() argument79 transpose_tensor = tensor.is_contiguous(); in prepare_matrix_for_cublas()85 transpose_tensor = false; in prepare_matrix_for_cublas()88 transpose_tensor = true; in prepare_matrix_for_cublas()[all …]