Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc71 /*indices=*/CreateTensor<int64>({2, 2}, {0, 0, 1, 1}), in TwoDimsSparseTensorSliceDatasetParams()
72 /*values=*/CreateTensor<int32>({2}, {888, 999}), in TwoDimsSparseTensorSliceDatasetParams()
73 /*dense_shape=*/CreateTensor<int64>({2}, {2, 2}), in TwoDimsSparseTensorSliceDatasetParams()
80 /*indices=*/CreateTensor<int64>({2, 3}, {0, 0, 0, 1, 1, 1}), in ThreeDimsSparseTensorSliceDatasetParams()
81 /*values=*/CreateTensor<double>({2}, {888.0, 999.0}), in ThreeDimsSparseTensorSliceDatasetParams()
82 /*dense_shape=*/CreateTensor<int64>({3}, {2, 2, 2}), in ThreeDimsSparseTensorSliceDatasetParams()
89 /*indices=*/CreateTensor<int64>({2, 4}, {0, 0, 0, 0, 1, 1, 1, 1}), in FourDimsSparseTensorSliceDatasetParams()
90 /*values=*/CreateTensor<tstring>({2}, {"a", "b"}), in FourDimsSparseTensorSliceDatasetParams()
91 /*dense_shape=*/CreateTensor<int64>({4}, {3, 2, 2, 2}), in FourDimsSparseTensorSliceDatasetParams()
98 /*indices=*/CreateTensor<int64>({2, 5}, {0, 0, 0, 0, 0, 1, 1, 1, 1, 1}), in FiveDimsSparseTensorSliceDatasetParams()
[all …]
Dtensor_slice_dataset_op_test.cc30 CreateTensor<int64>(TensorShape({2}), {1, 2}), in PlainTensorSliceDatasetParams()
31 CreateTensor<int64>(TensorShape({2, 2}), {1, 2, 3, 4}), in PlainTensorSliceDatasetParams()
32 CreateTensor<uint32>(TensorShape({2}), {2, 3}), in PlainTensorSliceDatasetParams()
33 CreateTensor<uint32>(TensorShape({2, 2}), {2, 3, 4, 5}), in PlainTensorSliceDatasetParams()
34 CreateTensor<uint64>(TensorShape({2}), {3, 4}), in PlainTensorSliceDatasetParams()
35 CreateTensor<uint64>(TensorShape({2, 2}), {3, 4, 5, 6}), in PlainTensorSliceDatasetParams()
36 CreateTensor<double>(TensorShape({2, 1}), {37.0, 38.0}), in PlainTensorSliceDatasetParams()
37 CreateTensor<tstring>(TensorShape({2, 1}), {"a", "b"})}; in PlainTensorSliceDatasetParams()
44 CreateTensor<Variant>( in NestedTensorSliceDatasetParams()
46 {CreateTensor<double>(TensorShape({2, 2}), {1.0, 2.0, 3.0, 4.0}), in NestedTensorSliceDatasetParams()
[all …]
Dpadded_batch_dataset_op_test.cc53 CreateTensor<int64>(TensorShape({}), {batch_size_})); in GetInputTensors()
61 CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
107 /*components=*/{CreateTensor<int64>( in PaddedBatchDatasetParams1()
113 /*padded_shapes=*/{CreateTensor<int64>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams1()
114 /*padded_values=*/{CreateTensor<int64>(TensorShape{}, {1})}, in PaddedBatchDatasetParams1()
141 /*padded_shapes=*/{CreateTensor<int64>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams2()
142 /*padded_values=*/{CreateTensor<int64>(TensorShape{}, {1})}, in PaddedBatchDatasetParams2()
169 /*padded_shapes=*/{CreateTensor<int64>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams3()
170 /*padded_values=*/{CreateTensor<int64>(TensorShape{}, {1})}, in PaddedBatchDatasetParams3()
199 /*padded_shapes=*/{CreateTensor<int64>(TensorShape{1}, {3})}, in PaddedBatchDatasetParams4()
[all …]
Dconcatenate_dataset_op_test.cc47 {CreateTensor<int64>(TensorShape{2, 3}, {1, 2, 3, 4, 5, 6}), in DifferentShapeConcatenateDatasetParams()
48 CreateTensor<int64>(TensorShape{2, 2}, {7, 8, 9, 10})}, in DifferentShapeConcatenateDatasetParams()
52 {CreateTensor<int64>(TensorShape{2, 2}, {11, 12, 13, 14}), in DifferentShapeConcatenateDatasetParams()
53 CreateTensor<int64>(TensorShape{2, 1}, {15, 16})}, in DifferentShapeConcatenateDatasetParams()
94 {CreateTensor<int64>(TensorShape{3}, {1, 2, 3}), in GetNextTestCases()
95 CreateTensor<int64>(TensorShape{2}, {7, 8}), in GetNextTestCases()
96 CreateTensor<int64>(TensorShape{3}, {4, 5, 6}), in GetNextTestCases()
97 CreateTensor<int64>(TensorShape{2}, {9, 10}), in GetNextTestCases()
98 CreateTensor<int64>(TensorShape{2}, {11, 12}), in GetNextTestCases()
99 CreateTensor<int64>(TensorShape{1}, {15}), in GetNextTestCases()
[all …]
Dtensor_dataset_op_test.cc79 {CreateTensor<int64>(TensorShape({}), {1}), in PlainTensorDatasetParams()
80 CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3}), in PlainTensorDatasetParams()
81 CreateTensor<double>(TensorShape({}), {37.0}), in PlainTensorDatasetParams()
82 CreateTensor<tstring>(TensorShape({1, 2}), {"a", "b"})}, in PlainTensorDatasetParams()
89 {CreateTensor<Variant>(TensorShape({}), in NestedTensorDatasetParams()
90 {CreateTensor<double>(TensorShape({2, 2}), in NestedTensorDatasetParams()
92 CreateTensor<Variant>( in NestedTensorDatasetParams()
94 {CreateTensor<tstring>(TensorShape({1, 2}), {"a", "b"})}), in NestedTensorDatasetParams()
95 CreateTensor<int64>(TensorShape({1, 3}), {1, 2, 3})}, in NestedTensorDatasetParams()
101 /*expected_outputs=*/{CreateTensor<int64>(TensorShape({}), {1}), in GetNextTestCases()
[all …]
Drepeat_dataset_op_test.cc43 return {CreateTensor<int64>(TensorShape({}), {count_})}; in GetInputTensors()
70 /*components=*/{CreateTensor<int64>(TensorShape{2, 2}, {1, 2, 3, 4}), in FiniteRepeatDatasetParams()
71 CreateTensor<tstring>(TensorShape{2, 1}, {"a", "b"})}, in FiniteRepeatDatasetParams()
83 /*components=*/{CreateTensor<int64>(TensorShape{2, 2}, {1, 2, 3, 4}), in EmptyRepeatDatasetParams()
84 CreateTensor<tstring>(TensorShape{2, 1}, {"a", "b"})}, in EmptyRepeatDatasetParams()
96 /*components=*/{CreateTensor<int64>(TensorShape{2, 1}, {1, 2})}, in ForeverRepeatDatasetParams()
109 {CreateTensor<int64>(TensorShape{2}, {1, 2}), in GetNextTestCases()
110 CreateTensor<tstring>(TensorShape{1}, {"a"}), in GetNextTestCases()
111 CreateTensor<int64>(TensorShape{2}, {3, 4}), in GetNextTestCases()
112 CreateTensor<tstring>(TensorShape{1}, {"b"}), in GetNextTestCases()
[all …]
Dbatch_dataset_op_test.cc130 {CreateTensor<int64>(TensorShape({3}), {0, 1, 2}), in GetNextTestCases()
131 CreateTensor<int64>(TensorShape({3}), {3, 4, 5}), in GetNextTestCases()
132 CreateTensor<int64>(TensorShape({3}), {6, 7, 8}), in GetNextTestCases()
133 CreateTensor<int64>(TensorShape({1}), {9})}}, in GetNextTestCases()
260 {CreateTensor<int64>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
261 CreateTensor<int64>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
262 CreateTensor<int64>(TensorShape({3}), {6, 7, 8}), in IteratorSaveAndRestoreTestCases()
263 CreateTensor<int64>(TensorShape({1}), {9})}}, in IteratorSaveAndRestoreTestCases()
267 {CreateTensor<int64>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
268 CreateTensor<int64>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_batch_dataset_op_test.cc47 Tensor batch_size = CreateTensor<int64>(TensorShape({}), {batch_size_}); in GetInputTensors()
49 CreateTensor<int64>(TensorShape({}), {num_parallel_calls_}); in GetInputTensors()
51 CreateTensor<bool>(TensorShape({}), {drop_remainder_}); in GetInputTensors()
212 {CreateTensor<int64>(TensorShape({3}), {0, 1, 2}), in GetNextTestCases()
213 CreateTensor<int64>(TensorShape({3}), {3, 4, 5}), in GetNextTestCases()
214 CreateTensor<int64>(TensorShape({3}), {6, 7, 8}), in GetNextTestCases()
215 CreateTensor<int64>(TensorShape({1}), {9})}}, in GetNextTestCases()
370 {CreateTensor<int64>(TensorShape({3}), {0, 1, 2}), in IteratorSaveAndRestoreTestCases()
371 CreateTensor<int64>(TensorShape({3}), {3, 4, 5}), in IteratorSaveAndRestoreTestCases()
372 CreateTensor<int64>(TensorShape({3}), {6, 7, 8}), in IteratorSaveAndRestoreTestCases()
[all …]
Dmap_defun_op_test.cc125 /*arguments=*/{CreateTensor<int64>(TensorShape({3, 2}), in TestCase1()
137 {CreateTensor<int64>(TensorShape({3, 2}), {0, 2, 4, 6, 8, 10})}}; in TestCase1()
144 /*arguments=*/{CreateTensor<int64>(TensorShape({3, 2}), in TestCase2()
146 CreateTensor<int64>(TensorShape({3, 2}), in TestCase2()
158 {CreateTensor<int64>(TensorShape({3, 2}), {0, 11, 22, 33, 44, 55})}}; in TestCase2()
166 /*arguments=*/{CreateTensor<int64>(TensorShape({3, 2}), in TestCase3()
169 {CreateTensor<int64>(TensorShape({2}), {10, 100})}, in TestCase3()
179 {CreateTensor<int64>(TensorShape({3, 2}), {10, 101, 12, 103, 14, 105})}}; in TestCase3()
185 /*arguments=*/{CreateTensor<int64>(TensorShape({3, 2}), in InvalidOutputTypes()
188 {CreateTensor<int64>(TensorShape({2}), {10, 100})}, in InvalidOutputTypes()
[all …]
Dparallel_interleave_dataset_op_test.cc59 CreateTensor<int64>(TensorShape({}), {cycle_length_})); in GetInputTensors()
61 CreateTensor<int64>(TensorShape({}), {block_length_})); in GetInputTensors()
63 CreateTensor<int64>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
65 CreateTensor<int64>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()
67 CreateTensor<int64>(TensorShape({}), {num_parallel_calls_})); in GetInputTensors()
129 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams1()
154 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams2()
179 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams3()
204 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams4()
229 /*components=*/{CreateTensor<tstring>( in ParallelInterleaveDatasetParams5()
[all …]
Dprefetch_dataset_op_test.cc46 return {CreateTensor<int64>(TensorShape({}), {buffer_size_})}; in GetInputTensors()
82 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams1()
99 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams2()
116 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams3()
133 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams4()
150 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams5()
167 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in PrefetchDatasetParams6()
183 /*components=*/{CreateTensor<int64>(TensorShape{10, 1}, in InvalidBufferSizePrefetchDatasetParams()
Dinterleave_dataset_op_test.cc51 CreateTensor<int64>(TensorShape({}), {cycle_length_})); in GetInputTensors()
53 CreateTensor<int64>(TensorShape({}), {block_length_})); in GetInputTensors()
109 {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})}, in InterleaveDatasetParams1()
131 {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})}, in InterleaveDatasetParams2()
153 {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})}, in InterleaveDatasetParams3()
175 {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})}, in InterleaveDatasetParams4()
197 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams5()
220 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams6()
243 {CreateTensor<tstring>(TensorShape{3, 3, 1}, in InterleaveDatasetParams7()
266 {CreateTensor<int64>(TensorShape{3, 3, 1}, {0, 1, 2, 3, 4, 5, 6, 7, 8})}, in InterleaveDatasetParamsWithInvalidCycleLength()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op_test.cc51 inputs.emplace_back(CreateTensor<int64>(TensorShape({}), {batch_size_})); in GetInputTensors()
53 CreateTensor<int64>(TensorShape({}), {num_parallel_calls_})); in GetInputTensors()
54 inputs.emplace_back(CreateTensor<bool>(TensorShape({}), {drop_remainder_})); in GetInputTensors()
253 {CreateTensor<int64>(TensorShape({2}), {0, 8}), in GetNextTestCases()
254 CreateTensor<int64>(TensorShape({2}), {16, 24}), in GetNextTestCases()
255 CreateTensor<int64>(TensorShape({1}), {32})}}, in GetNextTestCases()
264 {CreateTensor<int64>(TensorShape({2}), {0, 8}), in GetNextTestCases()
265 CreateTensor<int64>(TensorShape({2}), {16, 24}), in GetNextTestCases()
266 CreateTensor<int64>(TensorShape({1}), {32})}}}; in GetNextTestCases()
387 {CreateTensor<int64>(TensorShape({2}), {0, 8}), in IteratorSaveAndRestoreTestCases()
[all …]
Dparallel_interleave_dataset_op_test.cc59 CreateTensor<int64>(TensorShape({}), {cycle_length_})); in GetInputTensors()
61 CreateTensor<int64>(TensorShape({}), {block_length_})); in GetInputTensors()
63 CreateTensor<int64>(TensorShape({}), {buffer_output_elements_})); in GetInputTensors()
65 CreateTensor<int64>(TensorShape({}), {prefetch_input_elements_})); in GetInputTensors()
127 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams1()
153 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams2()
179 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams3()
205 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in ParallelInterleaveDatasetParams4()
231 /*components=*/{CreateTensor<tstring>( in ParallelInterleaveDatasetParams5()
278 /*components=*/{CreateTensor<int64>(TensorShape{3, 3, 1}, in InvalidCycleLengthParams()
[all …]
Dunique_dataset_op_test.cc60 {CreateTensor<int64>(TensorShape{12, 1}, in NormalCaseParams()
71 {CreateTensor<int64>(TensorShape{11, 1}, in LastRecordIsDuplicateParams()
82 {CreateTensor<int64>(TensorShape{5, 1}, {1, 1, 1, 1, 1})}, in AllRecordsTheSameParams()
92 {CreateTensor<int64>(TensorShape{0, 1}, {})}, in EmptyInputParams()
102 {CreateTensor<tstring>( in StringParams()
118 CreateTensor<int64>(TensorShape{1, 1}, {1}), in TwoComponentsParams()
119 CreateTensor<int64>(TensorShape{1, 1}, {42}), in TwoComponentsParams()
143 {CreateTensor<float>(TensorShape{1, 1}, {3.14})}, in FP32Params()
Ddirected_interleave_dataset_op_test.cc84 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in AlternateInputsParams()
99 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in SelectExhaustedInputParams()
114 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 0, 0, 0, 0, 0})}, in OneInputDatasetParams()
128 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 0, 0, 0, 0, 0})}, in ZeroInputDatasetParams()
143 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in LargeNumInputDatasetsParams()
160 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in SmallNumInputDatasetsParams()
175 /*components=*/{CreateTensor<int32>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in InvalidSelectorOuputDataType()
190 /*components=*/{CreateTensor<int64>(TensorShape{6, 1}, in InvalidSelectorOuputShape()
206 /*components=*/{CreateTensor<int64>(TensorShape{6}, {2, 1, 0, 1, 0, 1})}, in InvalidSelectorValues()
221 /*components=*/{CreateTensor<int64>(TensorShape{6}, {0, 1, 0, 1, 0, 1})}, in InvalidInputDatasetsDataType()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dtranspose_conv_test.cc119 CreateTensor(output_shape, output_shape_dims), in TestTransposeConvFloat()
120 CreateTensor(filter_data, filter_dims), in TestTransposeConvFloat()
121 CreateTensor(input_data, input_dims), in TestTransposeConvFloat()
122 CreateTensor(bias_data, bias_dims), in TestTransposeConvFloat()
123 CreateTensor(output_data, output_dims), in TestTransposeConvFloat()
164 CreateTensor(output_shape, output_shape_dims), filter_tensor, in TestTransposeConvQuantized()
226 using tflite::testing::CreateTensor; in TF_LITE_MICRO_TEST()
245 CreateTensor(output_shape, output_shape_dims), in TF_LITE_MICRO_TEST()
246 CreateTensor(tflite::testing::kInputData, input_dims), in TF_LITE_MICRO_TEST()
247 CreateTensor(tflite::testing::kFilterData, filter_dims), in TF_LITE_MICRO_TEST()
[all …]
Dsplit_test.cc45 CreateTensor(axis_data, axis_dims), CreateTensor(input_data, input_dims), in TestSplitTwoOutputsFloat()
46 CreateTensor(output1_data, output1_dims), in TestSplitTwoOutputsFloat()
47 CreateTensor(output2_data, output2_dims)}; in TestSplitTwoOutputsFloat()
106 CreateTensor(axis_data, axis_dims), in TestSplitFourOutputsFloat()
107 CreateTensor(input_data, input_dims), in TestSplitFourOutputsFloat()
108 CreateTensor(output1_data, output1_dims), in TestSplitFourOutputsFloat()
109 CreateTensor(output2_data, output2_dims), in TestSplitFourOutputsFloat()
110 CreateTensor(output3_data, output1_dims), in TestSplitFourOutputsFloat()
111 CreateTensor(output4_data, output1_dims)}; in TestSplitFourOutputsFloat()
173 CreateTensor(axis_data, axis_dims), in TestSplitTwoOutputsQuantized()
[all …]
Dpack_test.cc63 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
64 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsFloat()
65 CreateTensor(output_data, output_dims)}; in TestPackTwoInputsFloat()
96 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
97 CreateTensor(input2_data, input2_dims), in TestPackThreeInputsFloat()
98 CreateTensor(input3_data, input3_dims), in TestPackThreeInputsFloat()
99 CreateTensor(output_data, output_dims)}; in TestPackThreeInputsFloat()
167 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsQuantized32()
168 CreateTensor(input2_data, input2_dims), in TestPackTwoInputsQuantized32()
169 CreateTensor(output_data, output_dims)}; in TestPackTwoInputsQuantized32()
Dzeros_like_test.cc36 CreateTensor(input_data, input_dims), in TestZerosLikeFloat()
37 CreateTensor(output_data, output_dims), in TestZerosLikeFloat()
68 CreateTensor(input_data, input_dims), in TestZerosLikeInt32()
69 CreateTensor(output_data, output_dims), in TestZerosLikeInt32()
100 CreateTensor(input_data, input_dims), in TestZerosLikeInt64()
101 CreateTensor(output_data, output_dims), in TestZerosLikeInt64()
Dunpack_test.cc44 CreateTensor(input_data, input_dims), in TestUnpackThreeOutputsFloat()
45 CreateTensor(output1_data, output1_dims), in TestUnpackThreeOutputsFloat()
46 CreateTensor(output2_data, output2_dims), in TestUnpackThreeOutputsFloat()
47 CreateTensor(output3_data, output3_dims)}; in TestUnpackThreeOutputsFloat()
105 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestUnpackOneOutputFloat()
106 CreateTensor(output_data, output_dims)}; in TestUnpackOneOutputFloat()
224 CreateTensor(input_data, input_dims), in TestUnpackThreeOutputsQuantized32()
225 CreateTensor(output1_data, output1_dims), in TestUnpackThreeOutputsQuantized32()
226 CreateTensor(output2_data, output2_dims), in TestUnpackThreeOutputsQuantized32()
227 CreateTensor(output3_data, output3_dims)}; in TestUnpackThreeOutputsQuantized32()
Dreshape_test.cc124 TfLiteTensor input_tensor = CreateTensor(input_data, input_dims); in TestReshape()
125 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshape()
126 TfLiteTensor output_tensor = CreateTensor(output_data, output_dims); in TestReshape()
147 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshapeQuantized()
216 auto input_tensor = tflite::testing::CreateTensor(input_data, input_dims); in TF_LITE_MICRO_TEST()
221 auto output_tensor = tflite::testing::CreateTensor(output_data, output_dims); in TF_LITE_MICRO_TEST()
329 using tflite::testing::CreateTensor; in TF_LITE_MICRO_TEST()
335 auto input_tensor = CreateTensor(input_data, input_dims); in TF_LITE_MICRO_TEST()
340 auto output_tensor = CreateTensor(output_data, output_dims); in TF_LITE_MICRO_TEST()
346 auto shape_tensor = tflite::testing::CreateTensor(shape_data, shape_dims); in TF_LITE_MICRO_TEST()
Dbatch_to_space_nd_test.cc80 CreateTensor(input_data, input_dims), in TestBatchToSpaceNdFloat()
81 CreateTensor(block_shape_data, block_shape_dims), in TestBatchToSpaceNdFloat()
82 CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdFloat()
83 CreateTensor(output_data, output_dims), in TestBatchToSpaceNdFloat()
110 tflite::testing::CreateTensor(block_shape_data, block_shape_dims), in TestBatchToSpaceNdQuantized()
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestBatchToSpaceNdQuantized()
Dspace_to_batch_nd_test.cc80 CreateTensor(input_data, input_dims), in TestSpaceToBatchNdFloat()
81 CreateTensor(block_shape_data, block_shape_dims), in TestSpaceToBatchNdFloat()
82 CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdFloat()
83 CreateTensor(output_data, output_dims), in TestSpaceToBatchNdFloat()
110 tflite::testing::CreateTensor(block_shape_data, block_shape_dims), in TestSpaceToBatchNdQuantized()
111 tflite::testing::CreateTensor(crops_data, crops_dims), in TestSpaceToBatchNdQuantized()
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.cc296 t = ConverterTraits<T>::CreateTensor(ctx, state->inferred_shape); in Convert()
323 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
325 return tensorflow::unwrap(ctx)->CreateTensor(DT_INT64, dim_sizes); in CreateTensor()
363 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
365 return tensorflow::unwrap(ctx)->CreateTensor(DT_UINT64, dim_sizes); in CreateTensor()
400 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
402 return tensorflow::unwrap(ctx)->CreateTensor(DT_INT32, dim_sizes); in CreateTensor()
507 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
509 return tensorflow::unwrap(ctx)->CreateTensor(DT_FLOAT, dim_sizes); in CreateTensor()
523 static AbstractTensorInterface* CreateTensor( in CreateTensor() function
[all …]

123456