/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
H A D | list_dataset_op_test.cc | 116 {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 D | map_and_batch_dataset_op_test.cc | 51 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 D | directed_interleave_dataset_op_test.cc | 89 /*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 D | parallel_interleave_dataset_op_test.cc | 59 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 D | unique_dataset_op_test.cc | 61 {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 D | sparse_tensor_slice_dataset_op_test.cc | 75 /*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 D | tensor_slice_dataset_op_test.cc | 34 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 D | padded_batch_dataset_op_test.cc | 53 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 D | repeat_dataset_op_test.cc | 47 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 D | concatenate_dataset_op_test.cc | 47 {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 D | tensor_dataset_op_test.cc | 82 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 D | parallel_interleave_dataset_op_test.cc | 64 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 D | parallel_batch_dataset_op_test.cc | 50 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 D | map_defun_op_test.cc | 125 /*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 D | batch_dataset_op_test.cc | 135 {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 D | parallel_filter_dataset_op_test.cc | 55 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 D | interleave_dataset_op_test.cc | 51 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 D | prefetch_dataset_op_test.cc | 46 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 D | fixed_length_record_dataset_op_test.cc | 50 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 D | LstmTestHelper.hpp | 104 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 D | UnidirectionalSequenceLstmTestHelper.hpp | 126 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 D | ConvolutionTestHelper.hpp | 97 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 D | py_seq_tensor.cc | 296 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 D | conv_2d_tester.cc | 322 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 D | depthwise_conv_2d_tester.cc | 292 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()
|