Searched refs:TakeDatasetParams (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | take_dataset_op_test.cc | 25 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 …]
|
| D | optimize_dataset_op_test.cc | 74 TakeDatasetParams(RangeDatasetParams(-3, 3, 1), in TEST_F()
|
| D | dataset_test_base.cc | 1120 std::vector<Tensor> TakeDatasetParams::GetInputTensors() const { in GetInputTensors()
1124 Status TakeDatasetParams::GetInputNames( in GetInputNames()
1130 Status TakeDatasetParams::GetAttributes(AttributeVector* attr_vector) const { in GetAttributes()
1136 string TakeDatasetParams::dataset_type() const { in dataset_type()
|
| D | dataset_test_base.h | 314 class TakeDatasetParams : public DatasetParams {
317 TakeDatasetParams(T input_dataset_params, int count, in TakeDatasetParams() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | assert_next_dataset_op_test.cc | 72 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()
|