Home
last modified time | relevance | path

Searched refs:TensorSliceDatasetParams (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc28 TensorSliceDatasetParams PlainTensorSliceDatasetParams() { in PlainTensorSliceDatasetParams()
42 TensorSliceDatasetParams NestedTensorSliceDatasetParams() { in NestedTensorSliceDatasetParams()
57 std::vector<GetNextTestCase<TensorSliceDatasetParams>> GetNextTestCases() { in GetNextTestCases()
97 GetNextTestCase<TensorSliceDatasetParams>> {};
152 std::vector<DatasetOutputDtypesTestCase<TensorSliceDatasetParams>>
160 DATASET_OUTPUT_DTYPES_TEST_P(TensorSliceDatasetOpTest, TensorSliceDatasetParams, in DATASET_OUTPUT_DTYPES_TEST_P() argument
163 std::vector<DatasetOutputShapesTestCase<TensorSliceDatasetParams>> in DATASET_OUTPUT_DTYPES_TEST_P()
171 DATASET_OUTPUT_SHAPES_TEST_P(TensorSliceDatasetOpTest, TensorSliceDatasetParams, in DATASET_OUTPUT_SHAPES_TEST_P() argument
174 std::vector<CardinalityTestCase<TensorSliceDatasetParams>> in DATASET_OUTPUT_SHAPES_TEST_P()
180 DATASET_CARDINALITY_TEST_P(TensorSliceDatasetOpTest, TensorSliceDatasetParams, in DATASET_CARDINALITY_TEST_P() argument
[all …]
Dpadded_batch_dataset_op_test.cc106 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PaddedBatchDatasetParams1()
125 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in PaddedBatchDatasetParams2()
129 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in PaddedBatchDatasetParams2()
153 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in PaddedBatchDatasetParams3()
157 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in PaddedBatchDatasetParams3()
183 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in PaddedBatchDatasetParams4()
187 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in PaddedBatchDatasetParams4()
211 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in PaddedBatchDatasetParams5()
215 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in PaddedBatchDatasetParams5()
239 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in PaddedBatchDatasetParams6()
[all …]
Dparallel_interleave_dataset_op_test.cc128 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams1()
153 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams2()
178 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams3()
203 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams4()
228 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams5()
253 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams6()
278 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams7()
303 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams8()
328 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams9()
353 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams10()
[all …]
Dconcatenate_dataset_op_test.cc27 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in SameShapeConcatenateDatasetParams()
31 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in SameShapeConcatenateDatasetParams()
45 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in DifferentShapeConcatenateDatasetParams()
50 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in DifferentShapeConcatenateDatasetParams()
65 auto tensor_slice_dataset_params_0 = TensorSliceDatasetParams( in DifferentDtypeConcatenateDatasetParams()
68 auto tensor_slice_dataset_params_1 = TensorSliceDatasetParams( in DifferentDtypeConcatenateDatasetParams()
Dprefetch_dataset_op_test.cc81 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams1()
98 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams2()
115 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams3()
132 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams4()
149 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams5()
166 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams6()
182 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidBufferSizePrefetchDatasetParams()
Dinterleave_dataset_op_test.cc107 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams1()
129 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams2()
151 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams3()
173 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams4()
195 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams5()
218 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams6()
241 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams7()
264 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParamsWithInvalidCycleLength()
286 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParamsWithInvalidBlockLength()
Dfilter_dataset_op_test.cc85 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FilterDatasetParams1()
102 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FilterDatasetParams2()
119 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams1()
138 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams2()
156 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams3()
Dcache_dataset_ops_test.cc100 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams1()
113 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams2()
125 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams3()
138 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams4()
Drepeat_dataset_op_test.cc69 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FiniteRepeatDatasetParams()
82 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in EmptyRepeatDatasetParams()
95 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ForeverRepeatDatasetParams()
Dflat_map_dataset_op_test.cc83 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FlatMapDatasetParams1()
106 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidFlatMapDatasetParams()
Ddataset_test_base.cc1067 TensorSliceDatasetParams::TensorSliceDatasetParams( in TensorSliceDatasetParams() function in tensorflow::data::TensorSliceDatasetParams
1073 std::vector<Tensor> TensorSliceDatasetParams::GetInputTensors() const { in GetInputTensors()
1077 Status TensorSliceDatasetParams::GetInputNames( in GetInputNames()
1087 Status TensorSliceDatasetParams::GetAttributes( in GetAttributes()
1094 DataTypeVector TensorSliceDatasetParams::TensorSliceDtypes( in TensorSliceDtypes()
1103 std::vector<PartialTensorShape> TensorSliceDatasetParams::TensorSliceShapes( in TensorSliceShapes()
1116 string TensorSliceDatasetParams::dataset_type() const { in dataset_type()
Ddataset_test_base.h286 class TensorSliceDatasetParams : public DatasetParams {
288 TensorSliceDatasetParams(std::vector<Tensor> components, string node_name);
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op_test.cc58 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in NormalCaseParams()
69 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in LastRecordIsDuplicateParams()
80 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in AllRecordsTheSameParams()
90 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in EmptyInputParams()
100 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in StringParams()
115 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in TwoComponentsParams()
131 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in NoInputParams()
141 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FP32Params()
Ddirected_interleave_dataset_op_test.cc83 auto selector_input_dataset_params = TensorSliceDatasetParams( in AlternateInputsParams()
98 auto selector_input_dataset_params = TensorSliceDatasetParams( in SelectExhaustedInputParams()
113 auto selector_input_dataset_params = TensorSliceDatasetParams( in OneInputDatasetParams()
127 auto selector_input_dataset_params = TensorSliceDatasetParams( in ZeroInputDatasetParams()
142 auto selector_input_dataset_params = TensorSliceDatasetParams( in LargeNumInputDatasetsParams()
159 auto selector_input_dataset_params = TensorSliceDatasetParams( in SmallNumInputDatasetsParams()
174 auto selector_input_dataset_params = TensorSliceDatasetParams( in InvalidSelectorOuputDataType()
189 auto selector_input_dataset_params = TensorSliceDatasetParams( in InvalidSelectorOuputShape()
205 auto selector_input_dataset_params = TensorSliceDatasetParams( in InvalidSelectorValues()
220 auto selector_input_dataset_params = TensorSliceDatasetParams( in InvalidInputDatasetsDataType()
Dparallel_interleave_dataset_op_test.cc126 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams1()
152 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams2()
178 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams3()
204 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams4()
230 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams5()
254 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in EmptyInputParams()
277 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidCycleLengthParams()
301 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidBlockLengthParams()
325 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidBufferOutputElementsParams()
349 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPrefetchInputElementsParams()