Searched refs:dense_tensor_b (Results 1 – 1 of 1) sorted by relevance
105 const Tensor& dense_tensor_b, int* rank, in ValidateInputs() argument107 if (sparse_matrix_a.dtype() != dense_tensor_b.dtype()) { in ValidateInputs()111 " vs. b.dtype == ", DataTypeString(dense_tensor_b.dtype())); in ValidateInputs()115 if (*rank != dense_tensor_b.dims()) { in ValidateInputs()117 " vs. ", dense_tensor_b.dims(), "."); in ValidateInputs()120 *batch_size = (*rank == 2) ? 1 : dense_tensor_b.dim_size(0); in ValidateInputs()130 dense_tensor_b.dim_size(this->transpose_b_ ? *rank - 1 : *rank - 2); in ValidateInputs()135 dense_tensor_b.shape().DebugString()); in ValidateInputs()