Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc37 class SparseTensorSliceDatasetOpTest : public DatasetOpsTestBase {
76 {{{DatasetOpsTestBase::CreateTensor<int64>({2, 2}, {0, 0, 1, 1})}, in TestCases()
77 {DatasetOpsTestBase::CreateTensor<int32>({2}, {888, 999})}, in TestCases()
78 {DatasetOpsTestBase::CreateTensor<int64>({2}, {2, 2})}}, in TestCases()
79 {{{DatasetOpsTestBase::CreateTensor<int64>({1, 1}, {0})}, in TestCases()
80 {DatasetOpsTestBase::CreateTensor<int32>({1}, {888})}, in TestCases()
81 {DatasetOpsTestBase::CreateTensor<int64>({1}, {2})}}, in TestCases()
82 {{DatasetOpsTestBase::CreateTensor<int64>({1, 1}, {1})}, in TestCases()
83 {DatasetOpsTestBase::CreateTensor<int32>({1}, {999})}, in TestCases()
84 {DatasetOpsTestBase::CreateTensor<int64>({1}, {2})}}}, in TestCases()
[all …]
Dconcatenate_dataset_op_test.cc25 class ConcatenateDatasetOpTest : public DatasetOpsTestBase {
83 {{DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
85 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
87 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
89 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, in TestCase1()
92 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in TestCase1()
93 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {5, 6}), in TestCase1()
94 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in TestCase1()
95 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {7, 8}), in TestCase1()
96 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {11, 12}), in TestCase1()
[all …]
Dtensor_slice_dataset_op_test.cc37 class TensorSliceDatasetOpTest : public DatasetOpsTestBase {
80 {{{DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2}), {1, 2}), in TestCases()
81 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2, 2}), in TestCases()
83 DatasetOpsTestBase::CreateTensor<double>(TensorShape({2, 1}), in TestCases()
85 DatasetOpsTestBase::CreateTensor<string>(TensorShape({2, 1}), in TestCases()
87 {{DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in TestCases()
88 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({2}), {1, 2}), in TestCases()
89 DatasetOpsTestBase::CreateTensor<double>(TensorShape({1}), {37.0}), in TestCases()
90 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1}), {"a"}), in TestCases()
91 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {2}), in TestCases()
[all …]
Dtake_dataset_op_test.cc22 class TakeDatasetOpTest : public DatasetOpsTestBase {
74 {DatasetOpsTestBase::CreateTensor<int64>( in TakeLessTestCase()
78 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {0}), in TakeLessTestCase()
79 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {1}), in TakeLessTestCase()
80 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {2}), in TakeLessTestCase()
81 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {3})}, in TakeLessTestCase()
92 {DatasetOpsTestBase::CreateTensor<int64>( in TakeMoreTestCase()
96 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {0}), in TakeMoreTestCase()
97 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {1}), in TakeMoreTestCase()
98 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{1}, {2}), in TakeMoreTestCase()
[all …]
Ddataset_test_base.cc21 Status DatasetOpsTestBase::ExpectEqual(const Tensor& a, const Tensor& b) { in ExpectEqual()
38 Status DatasetOpsTestBase::CreateTensorSliceDatasetKernel( in CreateTensorSliceDatasetKernel()
56 Status DatasetOpsTestBase::CreateTensorSliceDataset( in CreateTensorSliceDataset()
89 Status DatasetOpsTestBase::CreateOpKernel( in CreateOpKernel()
99 Status DatasetOpsTestBase::CreateDataset(OpKernel* kernel, in CreateDataset()
109 Status DatasetOpsTestBase::CreateIteratorContext( in CreateIteratorContext()
119 Status DatasetOpsTestBase::GetDatasetFromContext(OpKernelContext* context, in GetDatasetFromContext()
128 Status DatasetOpsTestBase::InitThreadPool(int thread_num) { in InitThreadPool()
138 Status DatasetOpsTestBase::InitFunctionLibraryRuntime( in InitFunctionLibraryRuntime()
172 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
[all …]
Dtensor_dataset_op_test.cc37 class TensorDatasetOpTest : public DatasetOpsTestBase {
82 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in PlainTensorsTestCase()
83 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensorsTestCase()
84 DatasetOpsTestBase::CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensorsTestCase()
85 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1, 2}), in PlainTensorsTestCase()
88 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape({}), {1}), in PlainTensorsTestCase()
89 DatasetOpsTestBase::CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensorsTestCase()
90 DatasetOpsTestBase::CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensorsTestCase()
91 DatasetOpsTestBase::CreateTensor<string>(TensorShape({1, 2}), in PlainTensorsTestCase()
106 {DatasetOpsTestBase::CreateTensor<Variant>( in NestedTensorsTestCase()
[all …]
Dzip_dataset_op_test.cc31 class ZipDatasetOpTest : public DatasetOpsTestBase {
93 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {0}), in TestCase1()
94 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {10}), in TestCase1()
95 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {1}), in TestCase1()
96 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {11}), in TestCase1()
97 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {2}), in TestCase1()
98 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {12})}, in TestCase1()
107 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {0}), in TestCase2()
108 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {10}), in TestCase2()
109 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{}, {1}), in TestCase2()
[all …]
Drepeat_dataset_op_test.cc25 class RepeatDatasetOpTest : public DatasetOpsTestBase {
77 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2, 2}, {1, 2, 3, 4}), in FiniteRepeatTestCase()
78 DatasetOpsTestBase::CreateTensor<string>(TensorShape{2, 1}, {"a", "b"})}, in FiniteRepeatTestCase()
81 {DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in FiniteRepeatTestCase()
82 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"a"}), in FiniteRepeatTestCase()
83 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in FiniteRepeatTestCase()
84 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"b"}), in FiniteRepeatTestCase()
85 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {1, 2}), in FiniteRepeatTestCase()
86 DatasetOpsTestBase::CreateTensor<string>(TensorShape{1}, {"a"}), in FiniteRepeatTestCase()
87 DatasetOpsTestBase::CreateTensor<int64>(TensorShape{2}, {3, 4}), in FiniteRepeatTestCase()
[all …]
Ddataset_test_base.h40 class DatasetOpsTestBase : public ::testing::Test {
42 DatasetOpsTestBase() in DatasetOpsTestBase() function
48 ~DatasetOpsTestBase() {} in ~DatasetOpsTestBase()
Drange_dataset_op_test.cc36 class RangeDatasetOpTest : public DatasetOpsTestBase {
Dmap_dataset_op_test.cc40 class MapDatasetOpTest : public DatasetOpsTestBase {