Searched defs:transpose_tensor (Results 1 – 1 of 1) sorted by relevance
58 …d<Tensor> inline prepare_matrix_for_cublas(const Tensor& tensor, bool& transpose_tensor, bool tran… in prepare_matrix_for_cublas()77 …MaybeOwned<Tensor> inline prepare_matrix_for_cublas(const Tensor& tensor, bool& transpose_tensor) { in prepare_matrix_for_cublas()127 …ed<Tensor> prepare_batch_matrix_for_cublas(const Tensor& tensor, bool& transpose_tensor, int64_t& … in prepare_batch_matrix_for_cublas()