Searched refs:produced_tensors (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 244 Status DatasetOpsTestBase::ExpectEqual(std::vector<Tensor> produced_tensors, in ExpectEqual() argument
247 if (produced_tensors.size() != expected_tensors.size()) { in ExpectEqual()
249 "The two tensor vectors have different size (", produced_tensors.size(), in ExpectEqual()
253 if (produced_tensors.empty()) return Status::OK(); in ExpectEqual()
254 if (produced_tensors[0].dtype() != expected_tensors[0].dtype()) { in ExpectEqual()
257 produced_tensors[0].dtype(), " v.s. ", expected_tensors[0].dtype(), in ExpectEqual()
262 const DataType& dtype = produced_tensors[0].dtype(); in ExpectEqual()
266 std::sort(produced_tensors.begin(), produced_tensors.end(), \ in ExpectEqual()
296 for (int i = 0; i < produced_tensors.size(); ++i) { in ExpectEqual()
297 TF_RETURN_IF_ERROR(DatasetOpsTestBase::ExpectEqual(produced_tensors[i], in ExpectEqual()
|
| D | dataset_test_base.h | 567 static Status ExpectEqual(std::vector<Tensor> produced_tensors,
|