Home
last modified time | relevance | path

Searched refs:CreateTensor (Results 1 – 25 of 167) sorted by relevance

1234567

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dlist_dataset_op_test.cc116 {CreateTensor<int64_t>(TensorShape({}), {1}), in PlainListDatasetParams()
117 CreateTensor<int64_t>(TensorShape({2}), {1, 2}), in PlainListDatasetParams()
118 CreateTensor<uint32>(TensorShape({}), {2}), in PlainListDatasetParams()
119 CreateTensor<uint32>(TensorShape({2}), {2, 3}), in PlainListDatasetParams()
120 CreateTensor<uint64>(TensorShape({}), {3}), in PlainListDatasetParams()
121 CreateTensor<uint64>(TensorShape({2}), {3, 4}), in PlainListDatasetParams()
122 CreateTensor<double>(TensorShape({1}), {37.0}), in PlainListDatasetParams()
123 CreateTensor<tstring>(TensorShape({1}), {"a"})}, in PlainListDatasetParams()
124 {CreateTensor<int64_t>(TensorShape({}), {2}), in PlainListDatasetParams()
125 CreateTensor<int64_t>(TensorShape({2}), {3, 4}), in PlainListDatasetParams()
[all …]
H A Dmap_and_batch_dataset_op_test.cc51 inputs.emplace_back(CreateTensor<int64_t>(TensorShape({}), {batch_size_})); in GetInputTensors()
53 CreateTensor<int64_t>(TensorShape({}), {num_parallel_calls_})); in GetInputTensors()
54 inputs.emplace_back(CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
253 {CreateTensor<int64_t>(TensorShape({2}), {0, 8}), in GetNextTestCases()
254 CreateTensor<int64_t>(TensorShape({2}), {16, 24}), in GetNextTestCases()
255 CreateTensor<int64_t>(TensorShape({1}), {32})}}, in GetNextTestCases()
264 {CreateTensor<int64_t>(TensorShape({2}), {0, 8}), in GetNextTestCases()
265 CreateTensor<int64_t>(TensorShape({2}), {16, 24}), in GetNextTestCases()
266 CreateTensor<int64_t>(TensorShape({1}), {32})}}}; in GetNextTestCases()
387 {CreateTensor<int64_t>(TensorShape({2}), {0, 8}), in IteratorSaveAndRestoreTestCases()
[all …]
H A Ddirected_interleave_dataset_op_test.cc89 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in AlternateInputsParams()
106 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in SelectExhaustedInputParams()
123 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in OneInputDatasetParams()
139 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in ZeroInputDatasetParams()
154 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in StopOnEmptyDatasetParams()
171 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in SkipEmptyDatasetParams()
188 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in EmptyInputDatasetParams()
207 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in LargeNumInputDatasetsParams()
226 /*components=*/{CreateTensor<int64_t>(TensorShape{6}, in SmallNumInputDatasetsParams()
243 /*components=*/{CreateTensor<int32>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in InvalidSelectorOuputDataType()
[all …]
H A Dparallel_interleave_dataset_op_test.cc59 CreateTensor<int64_t>(TensorShape({}), {cycle_length_})); in GetInputTensors()
61 CreateTensor<int64_t>(TensorShape({}), {block_length_})); in GetInputTensors()
63 CreateTensor<int64_t>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
65 CreateTensor<int64_t>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()
127 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams1()
153 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams2()
179 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams3()
205 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams4()
231 /*components=*/{CreateTensor<tstring>( in ParallelInterleaveDatasetParams5()
278 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InvalidCycleLengthParams()
[all …]
H A Dunique_dataset_op_test.cc61 {CreateTensor<int64_t>(TensorShape{12, 1}, in NormalCaseParams()
72 {CreateTensor<int64_t>(TensorShape{11, 1}, in LastRecordIsDuplicateParams()
83 {CreateTensor<int64_t>(TensorShape{5, 1}, {1, 1, 1, 1, 1})}, in AllRecordsTheSameParams()
93 {CreateTensor<int64_t>(TensorShape{0, 1}, {})}, in EmptyInputParams()
103 {CreateTensor<tstring>( in StringParams()
119 CreateTensor<int64_t>(TensorShape{1, 1}, {1}), in TwoComponentsParams()
120 CreateTensor<int64_t>(TensorShape{1, 1}, {42}), in TwoComponentsParams()
144 {CreateTensor<float>(TensorShape{1, 1}, {3.14})}, in FP32Params()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op_test.cc75 /*indices=*/CreateTensor<int64_t>({2, 2}, {0, 0, 1, 1}), in TwoDimsSparseTensorSliceDatasetParams()
76 /*values=*/CreateTensor<int32>({2}, {888, 999}), in TwoDimsSparseTensorSliceDatasetParams()
77 /*dense_shape=*/CreateTensor<int64_t>({2}, {2, 2}), in TwoDimsSparseTensorSliceDatasetParams()
84 /*indices=*/CreateTensor<int64_t>({2, 3}, {0, 0, 0, 1, 1, 1}), in ThreeDimsSparseTensorSliceDatasetParams()
85 /*values=*/CreateTensor<double>({2}, {888.0, 999.0}), in ThreeDimsSparseTensorSliceDatasetParams()
86 /*dense_shape=*/CreateTensor<int64_t>({3}, {2, 2, 2}), in ThreeDimsSparseTensorSliceDatasetParams()
93 /*indices=*/CreateTensor<int64_t>({2, 4}, {0, 0, 0, 0, 1, 1, 1, 1}), in FourDimsSparseTensorSliceDatasetParams()
94 /*values=*/CreateTensor<tstring>({2}, {"a", "b"}), in FourDimsSparseTensorSliceDatasetParams()
95 /*dense_shape=*/CreateTensor<int64_t>({4}, {3, 2, 2, 2}), in FourDimsSparseTensorSliceDatasetParams()
102 /*indices=*/CreateTensor<int64_t>({2, 5}, {0, 0, 0, 0, 0, 1, 1, 1, 1, 1}), in FiveDimsSparseTensorSliceDatasetParams()
[all …]
H A Dtensor_slice_dataset_op_test.cc34 CreateTensor<int64_t>(TensorShape({2}), {1, 2}), in PlainTensorSliceDatasetParams()
35 CreateTensor<int64_t>(TensorShape({2, 2}), {1, 2, 3, 4}), in PlainTensorSliceDatasetParams()
36 CreateTensor<uint32>(TensorShape({2}), {2, 3}), in PlainTensorSliceDatasetParams()
37 CreateTensor<uint32>(TensorShape({2, 2}), {2, 3, 4, 5}), in PlainTensorSliceDatasetParams()
38 CreateTensor<uint64>(TensorShape({2}), {3, 4}), in PlainTensorSliceDatasetParams()
39 CreateTensor<uint64>(TensorShape({2, 2}), {3, 4, 5, 6}), in PlainTensorSliceDatasetParams()
40 CreateTensor<double>(TensorShape({2, 1}), {37.0, 38.0}), in PlainTensorSliceDatasetParams()
41 CreateTensor<tstring>(TensorShape({2, 1}), {"a", "b"})}; in PlainTensorSliceDatasetParams()
48 CreateTensor<Variant>( in NestedTensorSliceDatasetParams()
50 {CreateTensor<double>(TensorShape({2, 2}), {1.0, 2.0, 3.0, 4.0}), in NestedTensorSliceDatasetParams()
[all …]
H A Dpadded_batch_dataset_op_test.cc53 CreateTensor<int64_t>(TensorShape({}), {batch_size_})); in GetInputTensors()
61 CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
107 /*components=*/{CreateTensor<int64_t>( in PaddedBatchDatasetParams1()
113 /*padded_shapes=*/{CreateTensor<int64_t>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams1()
114 /*padded_values=*/{CreateTensor<int64_t>(TensorShape{}, {1})}, in PaddedBatchDatasetParams1()
141 /*padded_shapes=*/{CreateTensor<int64_t>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams2()
142 /*padded_values=*/{CreateTensor<int64_t>(TensorShape{}, {1})}, in PaddedBatchDatasetParams2()
169 /*padded_shapes=*/{CreateTensor<int64_t>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams3()
170 /*padded_values=*/{CreateTensor<int64_t>(TensorShape{}, {1})}, in PaddedBatchDatasetParams3()
199 /*padded_shapes=*/{CreateTensor<int64_t>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams4()
[all …]
H A Drepeat_dataset_op_test.cc47 return {CreateTensor<int64_t>(TensorShape({}), {count_})}; in GetInputTensors()
75 /*components=*/{CreateTensor<int64_t>(TensorShape{2, 2}, {1, 2, 3, 4}), in FiniteRepeatDatasetParams()
76 CreateTensor<tstring>(TensorShape{2, 1}, {"a", "b"})}, in FiniteRepeatDatasetParams()
88 /*components=*/{CreateTensor<int64_t>(TensorShape{2, 2}, {1, 2, 3, 4}), in EmptyRepeatDatasetParams()
89 CreateTensor<tstring>(TensorShape{2, 1}, {"a", "b"})}, in EmptyRepeatDatasetParams()
101 /*components=*/{CreateTensor<int64_t>(TensorShape{2, 1}, {1, 2})}, in ForeverRepeatDatasetParams()
114 {CreateTensor<int64_t>(TensorShape{2}, {1, 2}), in GetNextTestCases()
115 CreateTensor<tstring>(TensorShape{1}, {"a"}), in GetNextTestCases()
116 CreateTensor<int64_t>(TensorShape{2}, {3, 4}), in GetNextTestCases()
117 CreateTensor<tstring>(TensorShape{1}, {"b"}), in GetNextTestCases()
[all …]
H A Dconcatenate_dataset_op_test.cc47 {CreateTensor<int64_t>(TensorShape{2, 3}, {1, 2, 3, 4, 5, 6}), in DifferentShapeConcatenateDatasetParams()
48 CreateTensor<int64_t>(TensorShape{2, 2}, {7, 8, 9, 10})}, in DifferentShapeConcatenateDatasetParams()
52 {CreateTensor<int64_t>(TensorShape{2, 2}, {11, 12, 13, 14}), in DifferentShapeConcatenateDatasetParams()
53 CreateTensor<int64_t>(TensorShape{2, 1}, {15, 16})}, in DifferentShapeConcatenateDatasetParams()
94 {CreateTensor<int64_t>(TensorShape{3}, {1, 2, 3}), in GetNextTestCases()
95 CreateTensor<int64_t>(TensorShape{2}, {7, 8}), in GetNextTestCases()
96 CreateTensor<int64_t>(TensorShape{3}, {4, 5, 6}), in GetNextTestCases()
97 CreateTensor<int64_t>(TensorShape{2}, {9, 10}), in GetNextTestCases()
98 CreateTensor<int64_t>(TensorShape{2}, {11, 12}), in GetNextTestCases()
99 CreateTensor<int64_t>(TensorShape{1}, {15}), in GetNextTestCases()
[all …]
H A Dtensor_dataset_op_test.cc82 return {CreateTensor<int64_t>(TensorShape({}), {1}), in PlainTensors()
83 CreateTensor<int64_t>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensors()
84 CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensors()
85 CreateTensor<tstring>(TensorShape({1, 2}), {"a", "b"})}; in PlainTensors()
97 {CreateTensor<Variant>(TensorShape({}), in NestedTensorDatasetParams()
98 {CreateTensor<double>(TensorShape({2, 2}), in NestedTensorDatasetParams()
100 CreateTensor<Variant>( in NestedTensorDatasetParams()
102 {CreateTensor<tstring>(TensorShape({1, 2}), {"a", "b"})}), in NestedTensorDatasetParams()
103 CreateTensor<int64_t>(TensorShape({1, 3}), {1, 2, 3})}, in NestedTensorDatasetParams()
112 {CreateTensor<Variant>(TensorShape({}), in GetNextTestCases()
[all …]
H A Dparallel_interleave_dataset_op_test.cc64 CreateTensor<int64_t>(TensorShape({}), {cycle_length_})); in GetInputTensors()
66 CreateTensor<int64_t>(TensorShape({}), {block_length_})); in GetInputTensors()
68 CreateTensor<int64_t>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
70 CreateTensor<int64_t>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()
72 CreateTensor<int64_t>(TensorShape({}), {num_parallel_calls_})); in GetInputTensors()
134 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams1()
159 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams2()
184 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams3()
209 /*components=*/{CreateTensor<int64_t>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams4()
234 /*components=*/{CreateTensor<tstring>( in ParallelInterleaveDatasetParams5()
[all …]
H A Dparallel_batch_dataset_op_test.cc50 Tensor batch_size = CreateTensor<int64_t>(TensorShape({}), {batch_size_}); in GetInputTensors()
52 CreateTensor<int64_t>(TensorShape({}), {num_parallel_calls_}); in GetInputTensors()
54 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
265 {CreateTensor<int64_t>(TensorShape({3}), {0, 1, 2}), in GetNextTestCases()
266 CreateTensor<int64_t>(TensorShape({3}), {3, 4, 5}), in GetNextTestCases()
267 CreateTensor<int64_t>(TensorShape({3}), {6, 7, 8}), in GetNextTestCases()
268 CreateTensor<int64_t>(TensorShape({1}), {9})}}, in GetNextTestCases()
434 {CreateTensor<int64_t>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
435 CreateTensor<int64_t>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
436 CreateTensor<int64_t>(TensorShape({3}), {6, 7, 8}), in IteratorSaveAndRestoreTestCases()
[all …]
H A Dmap_defun_op_test.cc125 /*arguments=*/{CreateTensor<int64_t>(TensorShape({3, 2}), in TestCase1()
137 {CreateTensor<int64_t>(TensorShape({3, 2}), {0, 2, 4, 6, 8, 10})}}; in TestCase1()
145 /*arguments=*/{CreateTensor<int64_t>(TensorShape({3, 2}), in TestCase2()
147 CreateTensor<int64_t>(TensorShape({3, 2}), in TestCase2()
159 {CreateTensor<int64_t>(TensorShape({3, 2}), {0, 11, 22, 33, 44, 55})}}; in TestCase2()
166 /*arguments=*/{CreateTensor<int64_t>(TensorShape({3, 2}), in TestCase3()
169 {CreateTensor<int64_t>(TensorShape({2}), {10, 100})}, in TestCase3()
179 {CreateTensor<int64_t>(TensorShape({3, 2}), in TestCase3()
186 /*arguments=*/{CreateTensor<int64_t>(TensorShape({3, 2}), in InvalidOutputTypes()
189 {CreateTensor<int64_t>(TensorShape({2}), {10, 100})}, in InvalidOutputTypes()
[all …]
H A Dbatch_dataset_op_test.cc135 {CreateTensor<int64_t>(TensorShape({3}), {0, 1, 2}), in GetNextTestCases()
136 CreateTensor<int64_t>(TensorShape({3}), {3, 4, 5}), in GetNextTestCases()
137 CreateTensor<int64_t>(TensorShape({3}), {6, 7, 8}), in GetNextTestCases()
138 CreateTensor<int64_t>(TensorShape({1}), {9})}}, in GetNextTestCases()
265 {CreateTensor<int64_t>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
266 CreateTensor<int64_t>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
267 CreateTensor<int64_t>(TensorShape({3}), {6, 7, 8}), in IteratorSaveAndRestoreTestCases()
268 CreateTensor<int64_t>(TensorShape({1}), {9})}}, in IteratorSaveAndRestoreTestCases()
272 {CreateTensor<int64_t>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
273 CreateTensor<int64_t>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
[all …]
H A Dparallel_filter_dataset_op_test.cc55 CreateTensor<int64_t>(TensorShape({}), {num_parallel_calls_})); in GetInputTensors()
101 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams1()
120 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams2()
139 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams3()
158 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams4()
177 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams5()
196 {CreateTensor<int64_t>(TensorShape{9, 1}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in ParallelFilterDatasetParams6()
215 {CreateTensor<int64_t>(TensorShape{0}, {})}, in InputHasNoElementParams()
234 {CreateTensor<int64_t>(TensorShape{3, 3}, {0, 0, 0, 3, 4, 5, 6, 7, 8})}, in InvalidPredFuncFilterDatasetParams1()
255 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InvalidPredFuncFilterDatasetParams2()
[all …]
H A Dinterleave_dataset_op_test.cc51 CreateTensor<int64_t>(TensorShape({}), {cycle_length_})); in GetInputTensors()
53 CreateTensor<int64_t>(TensorShape({}), {block_length_})); in GetInputTensors()
110 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InterleaveDatasetParams1()
133 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InterleaveDatasetParams2()
156 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InterleaveDatasetParams3()
179 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InterleaveDatasetParams4()
202 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams5()
225 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams6()
248 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams7()
271 {CreateTensor<int64_t>(TensorShape{3, 3, 1}, in InterleaveDatasetParamsWithInvalidCycleLength()
[all …]
H A Dprefetch_dataset_op_test.cc46 return {CreateTensor<int64_t>(TensorShape({}), {buffer_size_})}; in GetInputTensors()
81 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams1()
98 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams2()
115 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams3()
132 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams4()
149 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams5()
166 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in PrefetchDatasetParams6()
182 /*components=*/{CreateTensor<int64_t>(TensorShape{10, 1}, in InvalidBufferSizePrefetchDatasetParams()
H A Dfixed_length_record_dataset_op_test.cc50 CreateTensor<tstring>(TensorShape({num_files}), filenames_), in GetInputTensors()
51 CreateTensor<int64_t>(TensorShape({}), {header_bytes_}), in GetInputTensors()
52 CreateTensor<int64_t>(TensorShape({}), {record_bytes_}), in GetInputTensors()
53 CreateTensor<int64_t>(TensorShape({}), {footer_bytes_}), in GetInputTensors()
54 CreateTensor<int64_t>(TensorShape({}), {buffer_size_}), in GetInputTensors()
55 CreateTensor<tstring>(TensorShape({}), {ToString(compression_type_)})}; in GetInputTensors()
/aosp_15_r20/external/armnn/delegate/test/
H A DLstmTestHelper.hpp104 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
119 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
137 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
150 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
163 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
178 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
196 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
209 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
222 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
237 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateLstmTfLiteModel()
[all …]
H A DUnidirectionalSequenceLstmTestHelper.hpp126 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
141 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
160 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
174 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
188 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
203 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
222 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
236 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
250 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
266 tensors.push_back(CreateTensor(flatBufferBuilder, in CreateUnidirectionalSequenceLstmTfLiteModel()
[all …]
H A DConvolutionTestHelper.hpp97 tensors[0] = CreateTensor(flatBufferBuilder,
104 tensors[1] = CreateTensor(flatBufferBuilder,
117 tensors[2] = CreateTensor(flatBufferBuilder,
123 tensors[3] = CreateTensor(flatBufferBuilder,
347 tensors[0] = CreateTensor(flatBufferBuilder,
354 tensors[1] = CreateTensor(flatBufferBuilder,
367 tensors[2] = CreateTensor(flatBufferBuilder,
373 tensors[3] = CreateTensor(flatBufferBuilder,
564 tensors[0] = CreateTensor(flatBufferBuilder, in CreateTransposeConvTfLiteModel()
569 tensors[1] = CreateTensor(flatBufferBuilder, in CreateTransposeConvTfLiteModel()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dpy_seq_tensor.cc296 t = ConverterTraits<T>::CreateTensor(ctx, state->inferred_shape); in Convert()
324 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
326 return tensorflow::unwrap(ctx)->CreateTensor(DT_INT64, dim_sizes); in CreateTensor()
364 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
366 return tensorflow::unwrap(ctx)->CreateTensor(DT_UINT64, dim_sizes); in CreateTensor()
402 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
404 return tensorflow::unwrap(ctx)->CreateTensor(DT_INT32, dim_sizes); in CreateTensor()
508 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
510 return tensorflow::unwrap(ctx)->CreateTensor(DT_FLOAT, dim_sizes); in CreateTensor()
524 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dconv_2d_tester.cc322 CreateTensor(builder, in CreateTfLiteModel()
335 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
345 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
349 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
354 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
365 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
369 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
378 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
382 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
H A Ddepthwise_conv_2d_tester.cc292 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
296 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
301 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
329 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
335 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
339 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
348 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()
352 tensors.emplace_back(CreateTensor( in CreateTfLiteModel()

1234567