Searched refs:num_rows_tensor (Results 1 – 1 of 1) sorted by relevance
1363 const Tensor* num_rows_tensor = c->input_tensor(2); in MatrixDiagV2Shape() local1367 if (num_rows_tensor != nullptr) { in MatrixDiagV2Shape()1368 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_rows_tensor, &num_rows)); in MatrixDiagV2Shape()