Searched refs:VerifyTypesMatch (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.h | 52 Status VerifyTypesMatch(const DataTypeVector& expected,
|
| D | concatenate_dataset_op_test.cc | 237 TF_EXPECT_OK(VerifyTypesMatch(dataset_->output_dtypes(), in TEST_P()
281 TF_EXPECT_OK(VerifyTypesMatch(iterator_->output_dtypes(), in TEST_P()
|
| D | multi_device_iterator_ops.cc | 73 VerifyTypesMatch(output_types_, iterator->output_dtypes())); in Init()
468 VerifyTypesMatch(output_types_, resource->output_types())); in VerifyResource()
643 VerifyTypesMatch(output_types_, resource->output_types())); in Compute()
|
| D | zip_dataset_op_test.cc | 212 VerifyTypesMatch(dataset_->output_dtypes(), expected_output_dtypes)); in TEST_P()
274 VerifyTypesMatch(iterator_->output_dtypes(), expected_output_dtypes)); in TEST_P()
|
| D | iterator_ops.cc | 175 VerifyTypesMatch(output_dtypes_, new_state->iterator->output_dtypes())); in Restore()
247 VerifyTypesMatch(output_dtypes_, iterator->output_dtypes())); in SetIteratorFromDataset()
470 VerifyTypesMatch(output_dtypes_, resource->output_dtypes())); in VerifyResource()
861 VerifyTypesMatch(output_dtypes_, (*iterator)->output_dtypes())); in TryInit()
1106 OP_REQUIRES_OK(ctx, VerifyTypesMatch(output_dtypes_, in Compute()
|
| D | tensor_dataset_op.cc | 41 VerifyTypesMatch((*output)->output_dtypes(), output_types_)); in MakeDataset()
|
| D | tensor_slice_dataset_op.cc | 58 VerifyTypesMatch((*output)->output_dtypes(), output_types_)); in MakeDataset()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 231 VerifyTypesMatch(dataset->output_dtypes(), expected_output_dtypes)); in TEST_F()
362 VerifyTypesMatch(iterator->output_dtypes(), expected_output_dtypes)); in TEST_F()
|
| D | repeat_dataset_op_test.cc | 258 TF_EXPECT_OK(VerifyTypesMatch(repeat_dataset->output_dtypes(), in TEST_P()
390 TF_EXPECT_OK(VerifyTypesMatch(iterator->output_dtypes(), in TEST_P()
|
| D | take_dataset_op_test.cc | 264 TF_EXPECT_OK(VerifyTypesMatch(take_dataset->output_dtypes(), in TEST_P()
398 TF_EXPECT_OK(VerifyTypesMatch(iterator->output_dtypes(), in TEST_P()
|
| D | dataset_utils.cc | 108 Status VerifyTypesMatch(const DataTypeVector& expected, in VerifyTypesMatch() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | indexed_dataset_op.cc | 136 VerifyTypesMatch(output_dtypes_, dataset->output_dtypes())); in set_materialized_dataset()
312 VerifyTypesMatch(output_dtypes_, resource->output_dtypes())); in VerifyResource()
|