Home
last modified time | relevance | path

Searched refs:PartialTensorShape (Results 1 – 25 of 356) sorted by relevance

12345678910>>...15

/external/tensorflow/tensorflow/core/framework/
Dpartial_tensor_shape_test.cc29 const PartialTensorShape s; in TEST()
35 const PartialTensorShape s({10, 5}); in TEST()
63 const auto s4 = s.Concatenate(PartialTensorShape()); in TEST()
69 PartialTensorShape s({10, 5, 20}); in TEST()
83 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
87 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
92 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
96 EXPECT_TRUE(PartialTensorShape::IsValid(proto)); in TEST()
99 EXPECT_FALSE(PartialTensorShape::IsValid(proto)); in TEST()
103 EXPECT_FALSE(PartialTensorShape::IsValid(proto)); in TEST()
[all …]
Dtensor_shape.h38 class PartialTensorShape; variable
331 std::is_same<Shape, PartialTensorShape>::value;
386 operator const PartialTensorShape&() const; // NOLINT(runtime/explicit)
507 PartialTensorShape* out);
509 PartialTensorShape* out);
511 PartialTensorShape* out);
513 PartialTensorShape* out);
532 class PartialTensorShape : public TensorShapeBase<PartialTensorShape> {
534 PartialTensorShape() {} in PartialTensorShape() function
535 using TensorShapeBase<PartialTensorShape>::TensorShapeBase;
[all …]
Dtensor_shape.cc33 static_assert(sizeof(TensorShapeRep) == sizeof(PartialTensorShape),
738 const auto& shape = *static_cast<const PartialTensorShape*>(this); in DebugString()
845 MAKE_SHAPE(int32, PartialTensorShape) in MAKE_SHAPE()
846 MAKE_SHAPE(int64_t, PartialTensorShape) in MAKE_SHAPE()
861 PartialTensorShape PartialTensorShape::Concatenate(int64_t size) const { in Concatenate()
862 PartialTensorShape out = *this; in Concatenate()
867 Status PartialTensorShape::ConcatenateWithStatus( in ConcatenateWithStatus()
868 int64_t size, PartialTensorShape* out) const { in ConcatenateWithStatus()
869 out = const_cast<PartialTensorShape*>(this); in ConcatenateWithStatus()
873 PartialTensorShape PartialTensorShape::Concatenate( in Concatenate()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dinterleave_dataset_op_test.cc32 std::vector<PartialTensorShape> output_shapes, in InterleaveDatasetParams()
99 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
121 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams1()
125 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams1()
144 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams2()
148 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams2()
167 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams3()
171 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams3()
190 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams4()
194 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams4()
[all …]
Dparallel_batch_dataset_op_test.cc32 std::vector<PartialTensorShape> output_shapes, in ParallelBatchDatasetParams()
100 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams1()
115 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams2()
130 /*output_shapes=*/{PartialTensorShape({-1})}, in ParallelBatchDatasetParams3()
145 /*output_shapes=*/{PartialTensorShape({3})}, in ParallelBatchDatasetParams4()
160 /*output_shapes=*/{PartialTensorShape({12})}, in ParallelBatchDatasetParams5()
175 /*output_shapes=*/{PartialTensorShape({-1})}, in ParallelBatchDatasetParams6()
190 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams7()
204 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams8()
218 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams9()
[all …]
Dparallel_interleave_dataset_op_test.cc38 const std::vector<PartialTensorShape>& output_shapes, in ParallelInterleaveDatasetParams()
125 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
148 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams1()
152 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams1()
173 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams2()
177 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams2()
198 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams3()
202 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams3()
223 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams4()
227 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams4()
[all …]
Dpadded_batch_dataset_op_test.cc33 std::vector<PartialTensorShape> output_shapes, in PaddedBatchDatasetParams()
118 /*output_shapes=*/{PartialTensorShape({2, 3})}, in PaddedBatchDatasetParams1()
136 /*output_shapes=*/{PartialTensorShape({-1})}, in PaddedBatchDatasetParams2()
146 /*output_shapes=*/{PartialTensorShape({2, 3})}, in PaddedBatchDatasetParams2()
164 /*output_shapes=*/{PartialTensorShape({-1})}, in PaddedBatchDatasetParams3()
174 /*output_shapes=*/{PartialTensorShape({2, 3})}, in PaddedBatchDatasetParams3()
194 /*output_shapes=*/{PartialTensorShape({-1})}, in PaddedBatchDatasetParams4()
204 /*output_shapes=*/{PartialTensorShape({-1, 3})}, in PaddedBatchDatasetParams4()
222 /*output_shapes=*/{PartialTensorShape({-1})}, in PaddedBatchDatasetParams5()
232 /*output_shapes=*/{PartialTensorShape({-1, -1})}, in PaddedBatchDatasetParams5()
[all …]
Dbatch_dataset_op_test.cc37 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams1()
49 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams2()
61 /*output_shapes=*/{PartialTensorShape({-1})}, in BatchDatasetParams3()
73 /*output_shapes=*/{PartialTensorShape({3})}, in BatchDatasetParams4()
85 /*output_shapes=*/{PartialTensorShape({12})}, in BatchDatasetParams5()
97 /*output_shapes=*/{PartialTensorShape({-1})}, in BatchDatasetParams6()
109 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams7()
120 /*output_shapes=*/{PartialTensorShape({3})}, in InvalidBatchSizeBatchDatasetParams()
181 /*expected_output_shapes=*/{PartialTensorShape({4})}}, in DatasetOutputShapesTestCases()
183 /*expected_output_shapes=*/{PartialTensorShape({4})}}, in DatasetOutputShapesTestCases()
[all …]
Dsparse_tensor_slice_dataset_op_test.cc35 : DatasetParams({tvalues}, {PartialTensorShape({})}, in SparseTensorSliceDatasetParams()
223 /*expected_output_shapes=*/{PartialTensorShape({1, 1}), in DATASET_OUTPUT_DTYPES_TEST_P()
224 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
225 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
227 /*expected_output_shapes=*/{PartialTensorShape({1, 2}), in DATASET_OUTPUT_DTYPES_TEST_P()
228 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
229 PartialTensorShape({2})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
231 /*expected_output_shapes=*/{PartialTensorShape({1, 3}), in DATASET_OUTPUT_DTYPES_TEST_P()
232 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
233 PartialTensorShape({3})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
[all …]
Dtake_dataset_op_test.cc29 /*output_shapes=*/{PartialTensorShape({1})}, in TakeLessTakeDatasetParams()
38 /*output_shapes=*/{PartialTensorShape({1})}, in TakeMoreTakeDatasetParams()
47 /*output_shapes=*/{PartialTensorShape({1})}, in TakeAllTakeDatasetParams()
56 /*output_shapes=*/{PartialTensorShape({1})}, in TakeNothingTakeDatasetParams()
103 /*expected_output_shapes=*/{PartialTensorShape({})}}, in DatasetOutputShapesTestCases()
105 /*expected_output_shapes=*/{PartialTensorShape({})}}, in DatasetOutputShapesTestCases()
107 /*expected_output_shapes=*/{PartialTensorShape({})}}, in DatasetOutputShapesTestCases()
109 /*expected_output_shapes=*/{PartialTensorShape({})}}}; in DatasetOutputShapesTestCases()
138 /*expected_output_shapes=*/{PartialTensorShape({})}}, in IteratorOutputShapesTestCases()
140 /*expected_output_shapes=*/{PartialTensorShape({})}}, in IteratorOutputShapesTestCases()
[all …]
Dzip_dataset_op_test.cc30 std::vector<PartialTensorShape> output_shapes, in ZipDatasetParams()
78 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in ZipDatasetParams1()
89 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in ZipDatasetParams2()
134 /*expected_output_shapes=*/{PartialTensorShape({}), in DATASET_OUTPUT_DTYPES_TEST_P()
135 PartialTensorShape({})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
137 /*expected_output_shapes=*/{PartialTensorShape({}), in DATASET_OUTPUT_DTYPES_TEST_P()
138 PartialTensorShape({})}}}; in DATASET_OUTPUT_DTYPES_TEST_P()
168 /*expected_output_shapes=*/{PartialTensorShape({}), in ITERATOR_OUTPUT_DTYPES_TEST_P()
169 PartialTensorShape({})}}, in ITERATOR_OUTPUT_DTYPES_TEST_P()
171 /*expected_output_shapes=*/{PartialTensorShape({}), in ITERATOR_OUTPUT_DTYPES_TEST_P()
[all …]
Dshard_dataset_op_test.cc27 std::vector<PartialTensorShape> output_shapes, in ShardDatasetParams()
78 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams1()
89 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams2()
100 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams3()
111 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams4()
122 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams5()
134 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams6()
146 /*output_shapes=*/{PartialTensorShape({})}, in ShardDatasetParams7()
157 /*output_shapes=*/{PartialTensorShape({})}, in InvalidShardDatasetParamsWithNoElemForEachShard()
168 /*output_shapes=*/{PartialTensorShape({})}, in InvalidShardDatasetParams1()
[all …]
Drepeat_dataset_op_test.cc35 std::vector<PartialTensorShape> output_shapes, in RepeatDatasetParams()
82 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in FiniteRepeatDatasetParams()
95 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in EmptyRepeatDatasetParams()
107 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in ForeverRepeatDatasetParams()
213 /*expected_output_shapes=*/{PartialTensorShape({2}), in DATASET_OUTPUT_DTYPES_TEST_P()
214 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
216 /*expected_output_shapes=*/{PartialTensorShape({2}), in DATASET_OUTPUT_DTYPES_TEST_P()
217 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
219 /*expected_output_shapes=*/{PartialTensorShape({1})}}}; in DATASET_OUTPUT_DTYPES_TEST_P()
252 /*expected_output_shapes=*/{PartialTensorShape({2}), in ITERATOR_OUTPUT_DTYPES_TEST_P()
[all …]
Dfinalize_dataset_op_test.cc29 std::vector<PartialTensorShape> output_shapes, in FinalizeDatasetParams()
112 /*output_shapes=*/{PartialTensorShape({})}, in NoOptimizationOptionsParams()
123 /*output_shapes=*/{PartialTensorShape({})}, in MaxIntraOpParallelismOptionsParams()
133 /*output_shapes=*/{PartialTensorShape({})}, in PrivateThreadPoolOptionsParams()
143 /*output_shapes=*/{PartialTensorShape({})}, in ModelOptionsParams()
153 /*output_shapes=*/{PartialTensorShape({})}, in OptimizationsDefaultOptionsParams()
163 /*output_shapes=*/{PartialTensorShape({})}, in AllChainedDatasetsOptionsParams()
170 /*output_shapes=*/{PartialTensorShape({})}, in NoOptimizationFinalizeParams()
177 /*output_shapes=*/{PartialTensorShape({})}, in MaxIntraOpParallelismParams()
184 /*output_shapes=*/{PartialTensorShape({})}, in PrivateThreadPoolParams()
[all …]
Dparallel_filter_dataset_op_test.cc37 std::vector<PartialTensorShape> output_shapes, string node_name) in ParallelFilterDatasetParams()
112 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams1()
131 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams2()
150 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams3()
169 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams4()
188 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams5()
207 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelFilterDatasetParams6()
226 /*output_shapes=*/{PartialTensorShape({})}, in InputHasNoElementParams()
247 /*output_shapes=*/{PartialTensorShape({3, 1})}, in InvalidPredFuncFilterDatasetParams1()
268 /*output_shapes=*/{PartialTensorShape({3, 1})}, in InvalidPredFuncFilterDatasetParams2()
[all …]
Dcache_dataset_ops_test.cc35 std::vector<PartialTensorShape> output_shapes, in CacheDatasetParams()
113 /*output_shapes=*/{PartialTensorShape({3, 1})}, kNodeName); in CacheDatasetParams1()
125 /*output_shapes=*/{PartialTensorShape({})}, kNodeName); in CacheDatasetParams2()
137 /*output_shapes=*/{PartialTensorShape({3, 1})}, in CacheDatasetParams3()
149 /*output_shapes=*/{PartialTensorShape({})}, in CacheDatasetParams4()
229 /*expected_output_shapes=*/{PartialTensorShape({3, 1})}}, in DatasetOutputShapesTestCases()
231 /*expected_output_shapes=*/{PartialTensorShape({})}}, in DatasetOutputShapesTestCases()
234 /*expected_output_shapes=*/{PartialTensorShape({3, 1})}}, in DatasetOutputShapesTestCases()
236 /*expected_output_shapes=*/{PartialTensorShape({})}}}; in DatasetOutputShapesTestCases()
265 /*expected_output_shapes=*/{PartialTensorShape({3, 1})}}, in IteratorOutputShapesTestCases()
[all …]
Dmap_dataset_op_test.cc36 /*output_shapes=*/{PartialTensorShape({})}, in MapDatasetParams1()
48 /*output_shapes=*/{PartialTensorShape({})}, in MapDatasetParams1()
61 /*output_shapes=*/{PartialTensorShape({2})}, in MapDatasetParams2()
71 /*output_shapes=*/{PartialTensorShape({1})}, in MapDatasetParams2()
88 /*output_shapes=*/{PartialTensorShape({})}, in MapDatasetParams3()
130 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
154 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparallel_interleave_dataset_op_test.cc35 const std::vector<PartialTensorShape>& output_shapes, string node_name) in ParallelInterleaveDatasetParams()
116 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
141 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams1()
145 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams1()
167 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams2()
171 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams2()
193 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams3()
197 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams3()
219 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams4()
223 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams4()
[all …]
Ddirected_interleave_dataset_op_test.cc30 std::vector<PartialTensorShape> output_shapes, in DirectedInterleaveDatasetParams()
99 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in AlternateInputsParams()
116 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SelectExhaustedInputParams()
132 /*output_shapes=*/{PartialTensorShape({})}, in OneInputDatasetParams()
147 /*output_shapes=*/{PartialTensorShape({})}, in ZeroInputDatasetParams()
164 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in StopOnEmptyDatasetParams()
181 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SkipEmptyDatasetParams()
198 /*output_shapes=*/{PartialTensorShape({})}, in EmptyInputDatasetParams()
217 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in LargeNumInputDatasetsParams()
236 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SmallNumInputDatasetsParams()
[all …]
Dmap_and_batch_dataset_op_test.cc32 std::vector<PartialTensorShape> output_shapes, string node_name) in MapAndBatchDatasetParams()
120 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams1()
137 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams2()
155 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams3()
172 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams4()
190 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams5()
208 /*output_shapes=*/{PartialTensorShape({2})}, in MapAndBatchDatasetParams6()
224 /*output_shapes=*/{PartialTensorShape({2})}, in InvalidNumParallelCallsMapAndBatchDatasetParams()
240 /*output_shapes=*/{PartialTensorShape({2})}, in InvalidBatchSizeMapAndBatchDatasetParams()
295 {PartialTensorShape({2})}}, in DatasetOutputShapesTestCases()
[all …]
Dlmdb_dataset_op_test.cc37 std::vector<PartialTensorShape> output_shapes, in LMDBDatasetParams()
96 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SingleValidInput()
103 /*output_shapes*/ {PartialTensorShape({}), PartialTensorShape({})}, in TwoValidInputs()
110 /*output_shapes*/ {PartialTensorShape({}), PartialTensorShape({})}, in EmptyInput()
117 /*output_shapes*/ {PartialTensorShape({}), PartialTensorShape({})}, in InvalidPathAtStart()
125 /*output_shapes*/ {PartialTensorShape({}), PartialTensorShape({})}, in InvalidPathInMiddle()
233 /*expected_output_shapes=*/{PartialTensorShape({}), in DatasetOutputShapesTestCases()
234 PartialTensorShape({})}}}; in DatasetOutputShapesTestCases()
260 /*expected_output_shapes=*/{PartialTensorShape({}), in IteratorOutputShapesTestCases()
261 PartialTensorShape({})}}}; in IteratorOutputShapesTestCases()
Dunique_dataset_op_test.cc29 std::vector<PartialTensorShape> output_shapes) in UniqueDatasetParams()
66 /*output_shapes=*/{PartialTensorShape({1})}); in NormalCaseParams()
77 /*output_shapes=*/{PartialTensorShape({1})}); in LastRecordIsDuplicateParams()
87 /*output_shapes=*/{PartialTensorShape({1})}); in AllRecordsTheSameParams()
97 /*output_shapes=*/{PartialTensorShape({1})}); in EmptyInputParams()
110 /*output_shapes=*/{PartialTensorShape({1})}); in StringParams()
126 /*output_shapes=*/{PartialTensorShape({1}), PartialTensorShape({1})}); in TwoComponentsParams()
137 /*output_shapes=*/{PartialTensorShape({})}); in NoInputParams()
148 /*output_shapes=*/{PartialTensorShape({1})}); in FP32Params()
204 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
[all …]
Dassert_prev_dataset_op_test.cc53 std::vector<PartialTensorShape> output_shapes, in AssertPrevDatasetParams()
98 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams1()
105 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams1()
114 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams2()
122 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams2()
135 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams2WithAttrs()
144 /*output_shapes=*/{PartialTensorShape({})}, in AssertPrevDatasetParams2WithAttrs()
153 /*output_shapes=*/{PartialTensorShape({})}, in InvalidAssertPrevDatasetParams()
159 /*output_shapes=*/{PartialTensorShape({})}, in InvalidAssertPrevDatasetParams()
168 /*output_shapes=*/{PartialTensorShape({})}, in ShortAssertPrevDatasetParams()
[all …]
Dassert_next_dataset_op_test.cc31 std::vector<PartialTensorShape> output_shapes, in AssertNextDatasetParams()
76 /*output_shapes=*/{PartialTensorShape({})}, in AssertNextDatasetParams1()
82 /*output_shapes=*/{PartialTensorShape({})}, in AssertNextDatasetParams1()
91 /*output_shapes=*/{PartialTensorShape({})}, in AssertNextDatasetParams2()
98 /*output_shapes=*/{PartialTensorShape({})}, in AssertNextDatasetParams2()
107 /*output_shapes=*/{PartialTensorShape({})}, in InvalidAssertNextDatasetParams()
112 /*output_shapes=*/{PartialTensorShape({})}, in InvalidAssertNextDatasetParams()
121 /*output_shapes=*/{PartialTensorShape({})}, in ShortAssertNextDatasetParams()
128 /*output_shapes=*/{PartialTensorShape({})}, in ShortAssertNextDatasetParams()
166 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc55 std::map<string, std::vector<PartialTensorShape>> expected = { in TEST()
56 {"A", {PartialTensorShape({2, 3})}}, {"B", {PartialTensorShape({3})}}, in TEST()
57 {"C", {PartialTensorShape()}}, {"D", {PartialTensorShape({2, 3})}}, in TEST()
58 {"E", {PartialTensorShape()}}, {"F", {PartialTensorShape()}}, in TEST()
59 {"G", {PartialTensorShape()}}, in TEST()
118 std::map<string, std::vector<PartialTensorShape>> expected = { in TEST()
119 {"while/Identity", {PartialTensorShape()}}, in TEST()
120 {"while/add", {PartialTensorShape({})}}, in TEST()

12345678910>>...15