Searched refs:ConcatenateDatasetParams (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | concatenate_dataset_op_test.cc | 26 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 …]
|
D | padded_batch_dataset_op_test.cc | 133 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 …]
|
D | dataset_test_base.cc | 1140 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()
|
D | dataset_test_base.h | 344 class ConcatenateDatasetParams : public DatasetParams { 347 ConcatenateDatasetParams(T input_dataset_params_0, P input_dataset_params_1, in ConcatenateDatasetParams() function
|