Home
last modified time | relevance | path

Searched refs:TakeDatasetParams (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dtake_dataset_op_test.cc25 TakeDatasetParams TakeLessTakeDatasetParams() { in TakeLessTakeDatasetParams()
26 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeLessTakeDatasetParams()
34 TakeDatasetParams TakeMoreTakeDatasetParams() { in TakeMoreTakeDatasetParams()
35 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeMoreTakeDatasetParams()
43 TakeDatasetParams TakeAllTakeDatasetParams() { in TakeAllTakeDatasetParams()
44 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeAllTakeDatasetParams()
52 TakeDatasetParams TakeNothingTakeDatasetParams() { in TakeNothingTakeDatasetParams()
53 return TakeDatasetParams(RangeDatasetParams(0, 10, 1), in TakeNothingTakeDatasetParams()
60 std::vector<GetNextTestCase<TakeDatasetParams>> GetNextTestCases() { in GetNextTestCases()
78 ITERATOR_GET_NEXT_TEST_P(TakeDatasetOpTest, TakeDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument
[all …]
Doptimize_dataset_op_test.cc74 TakeDatasetParams(RangeDatasetParams(-3, 3, 1), in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_prev_dataset_op_test.cc94 TakeDatasetParams take_dataset_params = in AssertPrevDatasetParams1()
95 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertPrevDatasetParams1()
110 TakeDatasetParams take_dataset_params = in AssertPrevDatasetParams2()
111 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertPrevDatasetParams2()
127 TakeDatasetParams take_dataset_params = TakeDatasetParams( in AssertPrevDatasetParams2WithAttrs()
149 TakeDatasetParams take_dataset_params = in InvalidAssertPrevDatasetParams()
150 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidAssertPrevDatasetParams()
164 TakeDatasetParams take_dataset_params = in ShortAssertPrevDatasetParams()
165 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in ShortAssertPrevDatasetParams()
Dassert_next_dataset_op_test.cc72 TakeDatasetParams take_dataset_params = in AssertNextDatasetParams1()
73 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertNextDatasetParams1()
87 TakeDatasetParams take_dataset_params = in AssertNextDatasetParams2()
88 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in AssertNextDatasetParams2()
103 TakeDatasetParams take_dataset_params = in InvalidAssertNextDatasetParams()
104 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in InvalidAssertNextDatasetParams()
117 TakeDatasetParams take_dataset_params = in ShortAssertNextDatasetParams()
118 TakeDatasetParams(RangeDatasetParams(0, 10, 1), in ShortAssertNextDatasetParams()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc1149 std::vector<Tensor> TakeDatasetParams::GetInputTensors() const { in GetInputTensors()
1153 Status TakeDatasetParams::GetInputNames( in GetInputNames()
1159 Status TakeDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()
1166 string TakeDatasetParams::dataset_type() const { return "Take"; } in dataset_type()
Ddataset_test_base.h333 class TakeDatasetParams : public DatasetParams {
336 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams() function