/external/tensorflow/tensorflow/core/kernels/data/ |
D | sparse_tensor_slice_dataset_op_test.cc | 71 /*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 …]
|
D | tensor_slice_dataset_op_test.cc | 30 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 …]
|
D | padded_batch_dataset_op_test.cc | 53 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 …]
|
D | concatenate_dataset_op_test.cc | 47 {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 …]
|
D | tensor_dataset_op_test.cc | 79 {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 …]
|
D | repeat_dataset_op_test.cc | 43 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 …]
|
D | batch_dataset_op_test.cc | 130 {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 …]
|
D | parallel_batch_dataset_op_test.cc | 47 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 …]
|
D | map_defun_op_test.cc | 125 /*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 …]
|
D | parallel_interleave_dataset_op_test.cc | 59 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 …]
|
D | prefetch_dataset_op_test.cc | 46 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()
|
D | interleave_dataset_op_test.cc | 51 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/ |
D | map_and_batch_dataset_op_test.cc | 51 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 …]
|
D | parallel_interleave_dataset_op_test.cc | 59 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 …]
|
D | unique_dataset_op_test.cc | 60 {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()
|
D | directed_interleave_dataset_op_test.cc | 84 /*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/ |
D | transpose_conv_test.cc | 119 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 …]
|
D | split_test.cc | 45 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 …]
|
D | pack_test.cc | 63 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()
|
D | zeros_like_test.cc | 36 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()
|
D | unpack_test.cc | 44 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()
|
D | reshape_test.cc | 124 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()
|
D | batch_to_space_nd_test.cc | 80 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()
|
D | space_to_batch_nd_test.cc | 80 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/ |
D | py_seq_tensor.cc | 296 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 …]
|