Searched refs:DatasetOpsTestBase (Results 1 – 25 of 50) sorted by relevance
12
197 DatasetOpsTestBase::DatasetOpsTestBase() in DatasetOpsTestBase() function in tensorflow::data::DatasetOpsTestBase205 DatasetOpsTestBase::~DatasetOpsTestBase() { in ~DatasetOpsTestBase()211 Status DatasetOpsTestBase::ExpectEqual(const Tensor& a, const Tensor& b) { in ExpectEqual()239 Status DatasetOpsTestBase::ExpectEqual(std::vector<Tensor> produced_tensors, in ExpectEqual()292 TF_RETURN_IF_ERROR(DatasetOpsTestBase::ExpectEqual(produced_tensors[i], in ExpectEqual()298 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel()323 Status DatasetOpsTestBase::CreateDatasetContext( in CreateDatasetContext()337 Status DatasetOpsTestBase::CreateDataset(OpKernel* kernel, in CreateDataset()347 Status DatasetOpsTestBase::RestoreIterator( in RestoreIterator()355 Status DatasetOpsTestBase::CreateIteratorContext( in CreateIteratorContext()[all …]
40 : public DatasetOpsTestBase,
100 class SaveDatasetV2OpTest : public DatasetOpsTestBase {103 TF_RETURN_IF_ERROR(DatasetOpsTestBase::Initialize(dataset_params)); in Initialize()
74 class SamplingDatasetOpTest : public DatasetOpsTestBase {};
82 class RandomDatasetOpTest : public DatasetOpsTestBase {};
69 class AssertNextDatasetOpTest : public DatasetOpsTestBase {};
56 class UniqueDatasetOpTest : public DatasetOpsTestBase {};
65 class LMDBDatasetOpTest : public DatasetOpsTestBase {};
65 class RewriteDatasetOpTest : public DatasetOpsTestBase {};
70 class OptimizeDatasetOpTest : public DatasetOpsTestBase {};
62 class GetOptionsOpTest : public DatasetOpsTestBase {};
72 class CacheDatasetOpTest : public DatasetOpsTestBase {75 TF_RETURN_IF_ERROR(DatasetOpsTestBase::Initialize(dataset_params)); in Initialize()
30 class OptionsDatasetOpTest : public DatasetOpsTestBase {};
44 class IteratorOpsTest : public DatasetOpsTestBase {
25 class MapDatasetOpTest : public DatasetOpsTestBase {};
22 class TakeDatasetOpTest : public DatasetOpsTestBase {};
23 class RangeDatasetOpTest : public DatasetOpsTestBase {};
70 class ZipDatasetOpTest : public DatasetOpsTestBase {};
79 class ConcatenateDatasetOpTest : public DatasetOpsTestBase {};
63 class SkipDatasetOpTest : public DatasetOpsTestBase {};
95 class ReduceDatasetOpTest : public DatasetOpsTestBase {};
80 class FlatMapDatasetOpTest : public DatasetOpsTestBase {};
23 class PrefetchDatasetOpTest : public DatasetOpsTestBase {};
73 class TextLineDatasetOpTest : public DatasetOpsTestBase {};
69 class ShardDatasetOpTest : public DatasetOpsTestBase {};