Home
last modified time | relevance | path

Searched refs:DatasetOpsTestBase (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.cc202 DatasetOpsTestBase::DatasetOpsTestBase() in DatasetOpsTestBase() function in tensorflow::data::DatasetOpsTestBase
210 DatasetOpsTestBase::~DatasetOpsTestBase() { in ~DatasetOpsTestBase()
216 Status DatasetOpsTestBase::ExpectEqual(const Tensor& a, const Tensor& b) { in ExpectEqual()
244 Status DatasetOpsTestBase::ExpectEqual(std::vector<Tensor> produced_tensors, in ExpectEqual()
297 TF_RETURN_IF_ERROR(DatasetOpsTestBase::ExpectEqual(produced_tensors[i], in ExpectEqual()
303 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel()
318 Status DatasetOpsTestBase::CreateDatasetContext( in CreateDatasetContext()
332 Status DatasetOpsTestBase::CreateDataset(OpKernel* kernel, in CreateDataset()
342 Status DatasetOpsTestBase::RestoreIterator( in RestoreIterator()
350 Status DatasetOpsTestBase::CreateIteratorContext( in CreateIteratorContext()
[all …]
Doptimize_dataset_op_test.cc70 class OptimizeDatasetOpTest : public DatasetOpsTestBase {};
Dcache_dataset_ops_test.cc67 class CacheDatasetOpTest : public DatasetOpsTestBase {
70 TF_RETURN_IF_ERROR(DatasetOpsTestBase::Initialize(dataset_params)); in Initialize()
Dmap_dataset_op_test.cc25 class MapDatasetOpTest : public DatasetOpsTestBase {};
Dtake_dataset_op_test.cc22 class TakeDatasetOpTest : public DatasetOpsTestBase {};
Drange_dataset_op_test.cc23 class RangeDatasetOpTest : public DatasetOpsTestBase {};
Dreduce_dataset_op_test.cc94 class ReduceDatasetOpTest : public DatasetOpsTestBase {};
Dskip_dataset_op_test.cc62 class SkipDatasetOpTest : public DatasetOpsTestBase {};
Dconcatenate_dataset_op_test.cc79 class ConcatenateDatasetOpTest : public DatasetOpsTestBase {};
Dzip_dataset_op_test.cc70 class ZipDatasetOpTest : public DatasetOpsTestBase {};
Dbatch_dataset_op_test.cc22 class BatchDatasetOpTest : public DatasetOpsTestBase {};
Dflat_map_dataset_op_test.cc79 class FlatMapDatasetOpTest : public DatasetOpsTestBase {};
Dprefetch_dataset_op_test.cc23 class PrefetchDatasetOpTest : public DatasetOpsTestBase {};
Dtext_line_dataset_op_test.cc72 class TextLineDatasetOpTest : public DatasetOpsTestBase {};
Dfilter_dataset_op_test.cc81 class FilterDatasetOpTest : public DatasetOpsTestBase {};
Dtf_record_dataset_op_test.cc66 class TFRecordDatasetOpTest : public DatasetOpsTestBase {};
Dshard_dataset_op_test.cc69 class ShardDatasetOpTest : public DatasetOpsTestBase {};
/external/tensorflow/tensorflow/core/data/
Dcompression_utils_test.cc25 : public DatasetOpsTestBase,
/external/tensorflow/tensorflow/core/data/service/
Dtest_util_test.cc50 TF_EXPECT_OK(DatasetOpsTestBase::ExpectEqual(result[i], test_case.output[i], in TEST()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op_test.cc78 class AutoShardDatasetOpTest : public DatasetOpsTestBase {};
Dsampling_dataset_op_test.cc73 class SamplingDatasetOpTest : public DatasetOpsTestBase {};
Dassert_next_dataset_op_test.cc69 class AssertNextDatasetOpTest : public DatasetOpsTestBase {};
Drandom_dataset_op_test.cc80 class RandomDatasetOpTest : public DatasetOpsTestBase {};
Dunique_dataset_op_test.cc55 class UniqueDatasetOpTest : public DatasetOpsTestBase {};
Dlmdb_dataset_op_test.cc65 class LMDBDatasetOpTest : public DatasetOpsTestBase {};

12