Searched refs:true_classes_shape (Results 1 – 1 of 1) sorted by relevance
33 ShapeHandle true_classes_shape; in CandidateSamplerShapeFn() local34 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 2, &true_classes_shape)); in CandidateSamplerShapeFn()35 DimensionHandle batch_size = c->Dim(true_classes_shape, 0); in CandidateSamplerShapeFn()