Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc32 TensorSliceDatasetParams PlainTensorSliceDatasetParams() { in PlainTensorSliceDatasetParams()
46 TensorSliceDatasetParams NestedTensorSliceDatasetParams() { in NestedTensorSliceDatasetParams()
61 std::vector<GetNextTestCase<TensorSliceDatasetParams>> GetNextTestCases() { in GetNextTestCases()
101 GetNextTestCase<TensorSliceDatasetParams>> {};
159 std::vector<DatasetOutputDtypesTestCase<TensorSliceDatasetParams>>
167 DATASET_OUTPUT_DTYPES_TEST_P(TensorSliceDatasetOpTest, TensorSliceDatasetParams, in DATASET_OUTPUT_DTYPES_TEST_P() argument
170 std::vector<DatasetOutputShapesTestCase<TensorSliceDatasetParams>> in DATASET_OUTPUT_DTYPES_TEST_P()
178 DATASET_OUTPUT_SHAPES_TEST_P(TensorSliceDatasetOpTest, TensorSliceDatasetParams, in DATASET_OUTPUT_SHAPES_TEST_P() argument
181 std::vector<CardinalityTestCase<TensorSliceDatasetParams>> in DATASET_OUTPUT_SHAPES_TEST_P()
187 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.cc133 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams1()
158 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams2()
183 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams3()
208 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams4()
233 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams5()
258 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams6()
283 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams7()
308 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams8()
333 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams9()
358 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelInterleaveDatasetParams10()
[all …]
Dparallel_filter_dataset_op_test.cc99 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams1()
118 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams2()
137 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams3()
156 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams4()
175 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams5()
194 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ParallelFilterDatasetParams6()
213 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InputHasNoElementParams()
232 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams1()
253 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams2()
274 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams3()
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.cc80 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams1()
97 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams2()
114 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams3()
131 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams4()
148 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams5()
165 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in PrefetchDatasetParams6()
181 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidBufferSizePrefetchDatasetParams()
Dinterleave_dataset_op_test.cc108 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams1()
131 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams2()
154 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams3()
177 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams4()
200 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams5()
223 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams6()
246 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParams7()
269 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParamsWithInvalidCycleLength()
292 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InterleaveDatasetParamsWithInvalidBlockLength()
Dfilter_dataset_op_test.cc86 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FilterDatasetParams1()
103 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FilterDatasetParams2()
120 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams1()
139 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams2()
158 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidPredFuncFilterDatasetParams3()
Dcache_dataset_ops_test.cc105 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams1()
118 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams2()
130 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams3()
143 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in CacheDatasetParams4()
Drepeat_dataset_op_test.cc74 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FiniteRepeatDatasetParams()
87 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in EmptyRepeatDatasetParams()
100 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in ForeverRepeatDatasetParams()
Dflat_map_dataset_op_test.cc84 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FlatMapDatasetParams1()
107 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in InvalidFlatMapDatasetParams()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op_test.cc88 auto selector_input_dataset_params = TensorSliceDatasetParams( in AlternateInputsParams()
105 auto selector_input_dataset_params = TensorSliceDatasetParams( in SelectExhaustedInputParams()
122 auto selector_input_dataset_params = TensorSliceDatasetParams( in OneInputDatasetParams()
138 auto selector_input_dataset_params = TensorSliceDatasetParams( in ZeroInputDatasetParams()
153 auto selector_input_dataset_params = TensorSliceDatasetParams( in StopOnEmptyDatasetParams()
170 auto selector_input_dataset_params = TensorSliceDatasetParams( in SkipEmptyDatasetParams()
187 auto selector_input_dataset_params = TensorSliceDatasetParams( in EmptyInputDatasetParams()
206 auto selector_input_dataset_params = TensorSliceDatasetParams( in LargeNumInputDatasetsParams()
225 auto selector_input_dataset_params = TensorSliceDatasetParams( in SmallNumInputDatasetsParams()
242 auto selector_input_dataset_params = TensorSliceDatasetParams( in InvalidSelectorOuputDataType()
[all …]
Dunique_dataset_op_test.cc59 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in NormalCaseParams()
70 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in LastRecordIsDuplicateParams()
81 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in AllRecordsTheSameParams()
91 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in EmptyInputParams()
101 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in StringParams()
116 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in TwoComponentsParams()
132 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in NoInputParams()
142 auto tensor_slice_dataset_params = TensorSliceDatasetParams( in FP32Params()
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()
Dassert_prev_dataset_op_test.cc128 TensorSliceDatasetParams( in AssertPrevDatasetParams2WithAttrs()
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc1095 TensorSliceDatasetParams::TensorSliceDatasetParams( in TensorSliceDatasetParams() function in tensorflow::data::TensorSliceDatasetParams
1102 std::vector<Tensor> TensorSliceDatasetParams::GetInputTensors() const { in GetInputTensors()
1106 Status TensorSliceDatasetParams::GetInputNames( in GetInputNames()
1115 Status TensorSliceDatasetParams::GetAttributes( in GetAttributes()
1125 DataTypeVector TensorSliceDatasetParams::TensorSliceDtypes( in TensorSliceDtypes()
1134 std::vector<PartialTensorShape> TensorSliceDatasetParams::TensorSliceShapes( in TensorSliceShapes()
1147 string TensorSliceDatasetParams::dataset_type() const { return "TensorSlice"; } in dataset_type()
Ddataset_test_base.h303 class TensorSliceDatasetParams : public DatasetParams {
305 TensorSliceDatasetParams(std::vector<Tensor> components, string node_name,