Home
last modified time | relevance | path

Searched refs:labels_batches (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1381 int64_t labels_batches = labels_ty.getDimSize(0); in Verify() local
1383 !ShapedType::isDynamic(labels_batches) && in Verify()
1384 features_batches != labels_batches) in Verify()
/external/tensorflow/tensorflow/python/kernel_tests/
Dmetrics_test.py1986 labels_batches = labels.reshape((batch_size, num_batches))
1996 tf_label = constant_op.constant(labels_batches[:, i])