Home
last modified time | relevance | path

Searched refs:ConcatenateDatasetParams (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op_test.cc26 ConcatenateDatasetParams SameShapeConcatenateDatasetParams() { in SameShapeConcatenateDatasetParams()
35 return ConcatenateDatasetParams( in SameShapeConcatenateDatasetParams()
44 ConcatenateDatasetParams DifferentShapeConcatenateDatasetParams() { in DifferentShapeConcatenateDatasetParams()
55 return ConcatenateDatasetParams( in DifferentShapeConcatenateDatasetParams()
64 ConcatenateDatasetParams DifferentDtypeConcatenateDatasetParams() { in DifferentDtypeConcatenateDatasetParams()
72 return ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in DifferentDtypeConcatenateDatasetParams()
81 std::vector<GetNextTestCase<ConcatenateDatasetParams>> GetNextTestCases() { in GetNextTestCases()
104 ITERATOR_GET_NEXT_TEST_P(ConcatenateDatasetOpTest, ConcatenateDatasetParams, in ITERATOR_GET_NEXT_TEST_P() argument
127 std::vector<DatasetOutputDtypesTestCase<ConcatenateDatasetParams>>
137 DATASET_OUTPUT_DTYPES_TEST_P(ConcatenateDatasetOpTest, ConcatenateDatasetParams, in DATASET_OUTPUT_DTYPES_TEST_P() argument
[all …]
Dpadded_batch_dataset_op_test.cc133 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParams2()
161 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParams3()
191 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParams4()
219 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParams5()
247 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParams6()
291 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParamsWithShortPaddingShape()
319 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParamsWithInvalidPaddingShape()
347 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParamsWithInvalidBatchSize()
376 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParamsWithInvalidPaddingShapesSize()
407 ConcatenateDatasetParams(std::move(tensor_slice_dataset_params_0), in PaddedBatchDatasetParamsWithInvalidPaddingValuesSize()
[all …]
Ddataset_test_base.cc1140 std::vector<Tensor> ConcatenateDatasetParams::GetInputTensors() const { in GetInputTensors()
1144 Status ConcatenateDatasetParams::GetInputNames( in GetInputNames()
1151 Status ConcatenateDatasetParams::GetAttributes( in GetAttributes()
1158 string ConcatenateDatasetParams::dataset_type() const { in dataset_type()
Ddataset_test_base.h344 class ConcatenateDatasetParams : public DatasetParams {
347 ConcatenateDatasetParams(T input_dataset_params_0, P input_dataset_params_1, in ConcatenateDatasetParams() function