Searched refs:num_cols_tensor (Results 1 – 1 of 1) sorted by relevance
1364 const Tensor* num_cols_tensor = c->input_tensor(3); in MatrixDiagV2Shape() local1370 if (num_cols_tensor != nullptr) { in MatrixDiagV2Shape()1371 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_cols_tensor, &num_cols)); in MatrixDiagV2Shape()