Home
last modified time | relevance | path

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

12345678910>>...12

/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.h40 class PartialTensorShape; variable
331 std::is_same<Shape, PartialTensorShape>::value;
363 operator const PartialTensorShape&() const; // NOLINT(runtime/explicit)
483 static Status MakeShape(const int32* dims, int64 n, PartialTensorShape* out);
484 static Status MakeShape(const int64* dims, int64 n, PartialTensorShape* out);
486 PartialTensorShape* out);
488 PartialTensorShape* out);
506 class PartialTensorShape : public TensorShapeBase<PartialTensorShape> {
508 PartialTensorShape() {} in PartialTensorShape() function
509 using TensorShapeBase<PartialTensorShape>::TensorShapeBase;
[all …]
Dtensor_shape.cc34 static_assert(sizeof(TensorShapeRep) == sizeof(PartialTensorShape),
726 const auto& shape = *static_cast<const PartialTensorShape*>(this); in DebugString()
833 MAKE_SHAPE(int32, PartialTensorShape) in MAKE_SHAPE()
834 MAKE_SHAPE(int64, PartialTensorShape) in MAKE_SHAPE()
849 PartialTensorShape PartialTensorShape::Concatenate(int64 size) const { in Concatenate()
850 PartialTensorShape out = *this; in Concatenate()
855 Status PartialTensorShape::ConcatenateWithStatus( in ConcatenateWithStatus()
856 int64 size, PartialTensorShape* out) const { in ConcatenateWithStatus()
857 out = const_cast<PartialTensorShape*>(this); in ConcatenateWithStatus()
861 PartialTensorShape PartialTensorShape::Concatenate( in Concatenate()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dinterleave_dataset_op_test.cc32 std::vector<PartialTensorShape> output_shapes, in InterleaveDatasetParams()
98 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
119 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams1()
123 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams1()
141 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams2()
145 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams2()
163 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams3()
167 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams3()
185 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in InterleaveDatasetParams4()
189 /*output_shapes=*/{PartialTensorShape({1})}, in InterleaveDatasetParams4()
[all …]
Dparallel_interleave_dataset_op_test.cc33 const std::vector<PartialTensorShape>& output_shapes, in ParallelInterleaveDatasetParams()
120 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
143 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams1()
147 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams1()
168 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams2()
172 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams2()
193 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams3()
197 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams3()
218 std::vector<PartialTensorShape>({PartialTensorShape({1})})), in ParallelInterleaveDatasetParams4()
222 /*output_shapes=*/{PartialTensorShape({1})}, in ParallelInterleaveDatasetParams4()
[all …]
Dparallel_batch_dataset_op_test.cc32 std::vector<PartialTensorShape> output_shapes, in ParallelBatchDatasetParams()
89 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams1()
101 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams2()
114 /*output_shapes=*/{PartialTensorShape({-1})}, in ParallelBatchDatasetParams3()
126 /*output_shapes=*/{PartialTensorShape({3})}, in ParallelBatchDatasetParams4()
139 /*output_shapes=*/{PartialTensorShape({12})}, in ParallelBatchDatasetParams5()
152 /*output_shapes=*/{PartialTensorShape({-1})}, in ParallelBatchDatasetParams6()
164 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams7()
175 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams8()
186 /*output_shapes=*/{PartialTensorShape({4})}, in ParallelBatchDatasetParams9()
[all …]
Dbatch_dataset_op_test.cc32 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams1()
44 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams2()
56 /*output_shapes=*/{PartialTensorShape({-1})}, in BatchDatasetParams3()
68 /*output_shapes=*/{PartialTensorShape({3})}, in BatchDatasetParams4()
80 /*output_shapes=*/{PartialTensorShape({12})}, in BatchDatasetParams5()
92 /*output_shapes=*/{PartialTensorShape({-1})}, in BatchDatasetParams6()
104 /*output_shapes=*/{PartialTensorShape({4})}, in BatchDatasetParams7()
115 /*output_shapes=*/{PartialTensorShape({3})}, in InvalidBatchSizeBatchDatasetParams()
176 /*expected_output_shapes=*/{PartialTensorShape({4})}}, in DatasetOutputShapesTestCases()
178 /*expected_output_shapes=*/{PartialTensorShape({4})}}, in DatasetOutputShapesTestCases()
[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 …]
Dsparse_tensor_slice_dataset_op_test.cc31 : DatasetParams({tvalues}, {PartialTensorShape({})}, in SparseTensorSliceDatasetParams()
219 /*expected_output_shapes=*/{PartialTensorShape({1, 1}), in DATASET_OUTPUT_DTYPES_TEST_P()
220 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
221 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
223 /*expected_output_shapes=*/{PartialTensorShape({1, 2}), in DATASET_OUTPUT_DTYPES_TEST_P()
224 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
225 PartialTensorShape({2})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
227 /*expected_output_shapes=*/{PartialTensorShape({1, 3}), in DATASET_OUTPUT_DTYPES_TEST_P()
228 PartialTensorShape({1}), in DATASET_OUTPUT_DTYPES_TEST_P()
229 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()
133 /*expected_output_shapes=*/{PartialTensorShape({}), in DATASET_OUTPUT_DTYPES_TEST_P()
134 PartialTensorShape({})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
136 /*expected_output_shapes=*/{PartialTensorShape({}), in DATASET_OUTPUT_DTYPES_TEST_P()
137 PartialTensorShape({})}}}; in DATASET_OUTPUT_DTYPES_TEST_P()
167 /*expected_output_shapes=*/{PartialTensorShape({}), in ITERATOR_OUTPUT_DTYPES_TEST_P()
168 PartialTensorShape({})}}, in ITERATOR_OUTPUT_DTYPES_TEST_P()
170 /*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.cc31 std::vector<PartialTensorShape> output_shapes, in RepeatDatasetParams()
77 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in FiniteRepeatDatasetParams()
90 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in EmptyRepeatDatasetParams()
102 /*output_shapes=*/{PartialTensorShape({2}), PartialTensorShape({1})}, in ForeverRepeatDatasetParams()
208 /*expected_output_shapes=*/{PartialTensorShape({2}), in DATASET_OUTPUT_DTYPES_TEST_P()
209 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
211 /*expected_output_shapes=*/{PartialTensorShape({2}), in DATASET_OUTPUT_DTYPES_TEST_P()
212 PartialTensorShape({1})}}, in DATASET_OUTPUT_DTYPES_TEST_P()
214 /*expected_output_shapes=*/{PartialTensorShape({1})}}}; in DATASET_OUTPUT_DTYPES_TEST_P()
247 /*expected_output_shapes=*/{PartialTensorShape({2}), in ITERATOR_OUTPUT_DTYPES_TEST_P()
[all …]
Dcache_dataset_ops_test.cc31 std::vector<PartialTensorShape> output_shapes, in CacheDatasetParams()
108 /*output_shapes=*/{PartialTensorShape({3, 1})}, kNodeName); in CacheDatasetParams1()
120 /*output_shapes=*/{PartialTensorShape({})}, kNodeName); in CacheDatasetParams2()
132 /*output_shapes=*/{PartialTensorShape({3, 1})}, in CacheDatasetParams3()
144 /*output_shapes=*/{PartialTensorShape({})}, in CacheDatasetParams4()
224 /*expected_output_shapes=*/{PartialTensorShape({3, 1})}}, in DatasetOutputShapesTestCases()
226 /*expected_output_shapes=*/{PartialTensorShape({})}}, in DatasetOutputShapesTestCases()
229 /*expected_output_shapes=*/{PartialTensorShape({3, 1})}}, in DatasetOutputShapesTestCases()
231 /*expected_output_shapes=*/{PartialTensorShape({})}}}; in DatasetOutputShapesTestCases()
260 /*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()
Dskip_dataset_op_test.cc27 std::vector<PartialTensorShape> output_shapes, in SkipDatasetParams()
70 /*output_shapes=*/{PartialTensorShape({1})}, in SkipDatasetParams1()
80 /*output_shapes=*/{PartialTensorShape({1})}, in SkipDatasetParams2()
90 /*output_shapes=*/{PartialTensorShape({1})}, in SkipDatasetParams3()
100 /*output_shapes=*/{PartialTensorShape({1})}, in SkipDatasetParams4()
110 /*output_shapes=*/{PartialTensorShape({1})}, in SkipDatasetParams5()
156 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({})})); in TEST_F()
184 TF_ASSERT_OK(CheckIteratorOutputShapes({PartialTensorShape({})})); in TEST_F()
Dfilter_dataset_op_test.cc31 std::vector<PartialTensorShape> output_shapes, in FilterDatasetParams()
96 /*output_shapes=*/{PartialTensorShape({1})}, in FilterDatasetParams1()
113 /*output_shapes=*/{PartialTensorShape({})}, in FilterDatasetParams2()
132 /*output_shapes=*/{PartialTensorShape({3, 1})}, in InvalidPredFuncFilterDatasetParams1()
150 /*output_shapes=*/{PartialTensorShape({3, 1})}, in InvalidPredFuncFilterDatasetParams2()
168 /*output_shapes=*/{PartialTensorShape({})}, in InvalidPredFuncFilterDatasetParams3()
205 /*expected_output_shapes=*/{PartialTensorShape({1})}}, in DatasetOutputShapesTestCases()
207 /*expected_output_shapes=*/{PartialTensorShape({})}}}; in DatasetOutputShapesTestCases()
232 /*expected_output_shapes=*/{PartialTensorShape({1})}}, in IteratorOutputShapesTestCases()
234 /*expected_output_shapes=*/{PartialTensorShape({})}}}; in IteratorOutputShapesTestCases()
/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.cc29 std::vector<PartialTensorShape> output_shapes, in DirectedInterleaveDatasetParams()
92 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in AlternateInputsParams()
107 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SelectExhaustedInputParams()
121 /*output_shapes=*/{PartialTensorShape({})}, in OneInputDatasetParams()
134 /*output_shapes=*/{PartialTensorShape({})}, in ZeroInputDatasetParams()
151 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in LargeNumInputDatasetsParams()
168 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in SmallNumInputDatasetsParams()
183 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in InvalidSelectorOuputDataType()
199 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in InvalidSelectorOuputShape()
214 /*output_shapes=*/{PartialTensorShape({}), PartialTensorShape({})}, in InvalidSelectorValues()
[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()
65 /*output_shapes=*/{PartialTensorShape({1})}); in NormalCaseParams()
76 /*output_shapes=*/{PartialTensorShape({1})}); in LastRecordIsDuplicateParams()
86 /*output_shapes=*/{PartialTensorShape({1})}); in AllRecordsTheSameParams()
96 /*output_shapes=*/{PartialTensorShape({1})}); in EmptyInputParams()
109 /*output_shapes=*/{PartialTensorShape({1})}); in StringParams()
125 /*output_shapes=*/{PartialTensorShape({1}), PartialTensorShape({1})}); in TwoComponentsParams()
136 /*output_shapes=*/{PartialTensorShape({})}); in NoInputParams()
147 /*output_shapes=*/{PartialTensorShape({1})}); in FP32Params()
203 TF_ASSERT_OK(CheckDatasetOutputShapes({PartialTensorShape({1})})); in TEST_F()
[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 …]
Dauto_shard_dataset_op_test.cc30 std::vector<PartialTensorShape> output_shapes, in AutoShardDatasetParams()
88 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams1()
100 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams2()
113 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams3()
128 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams4()
140 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams5()
152 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams6()
164 /*output_shapes=*/{PartialTensorShape({})}, in AutoShardDatasetParams7()
/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>>...12