Home
last modified time | relevance | path

Searched refs:input1_dims (Results 1 – 13 of 13) sorted by relevance

/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/
Dpack_test.cc53 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsFloat() local
61 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
85 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackThreeInputsFloat() local
94 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
117 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized() local
129 CreateQuantizedTensor(input1_data, input1_dims, 1.0, 128), in TestPackTwoInputsQuantized()
151 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized32() local
159 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsQuantized32()
Dmaximum_minimum_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinFloat() local
41 CreateTensor(input1_data, input1_dims), in TestMaxMinFloat()
71 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantized() local
80 CreateQuantizedTensor(input1_data, input1_dims, input1_scale, in TestMaxMinQuantized()
110 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantizedInt32() local
119 CreateTensor(input1_data, input1_dims), in TestMaxMinQuantizedInt32()
Dconcatenation_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateTwoInputs() local
39 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestConcatenateTwoInputs()
74 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateQuantizedTwoInputs() local
82 CreateQuantizedTensor(input1_data, input1_dims, input_scale, in TestConcatenateQuantizedTwoInputs()
Dmul_test.cc74 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulFloat() local
83 CreateTensor(input1_data, input1_dims), in TestMulFloat()
101 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulQuantized() local
110 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestMulQuantized()
Dlogical_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestLogicalOp() local
41 CreateTensor(input1_data, input1_dims), in TestLogicalOp()
Dfloor_div_test.cc46 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestFloorDiv() local
52 CreateTensor(input1_data, input1_dims), in TestFloorDiv()
Dfloor_mod_test.cc46 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestFloorMod() local
52 CreateTensor(input1_data, input1_dims), in TestFloorMod()
Dcomparisons_test.cc58 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonFloat() local
63 CreateTensor(input1_data, input1_dims), in TestComparisonFloat()
76 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonBool() local
81 CreateTensor(input1_data, input1_dims), in TestComparisonBool()
94 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonInt() local
99 CreateTensor(input1_data, input1_dims), in TestComparisonInt()
115 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedUInt8() local
120 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedUInt8()
138 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedInt8() local
143 CreateQuantizedTensor(input1_data, input1_quantized, input1_dims, in TestComparisonQuantizedInt8()
Dif_test.cc34 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestIf() local
43 CreateTensor(input1_data, input1_dims), in TestIf()
Dadd_test.cc92 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddFloat() local
100 CreateTensor(input1_data, input1_dims), in TestAddFloat()
119 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddQuantized() local
128 input1_dims, input1_scale, in TestAddQuantized()
Dsub_test.cc91 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubFloat() local
99 CreateTensor(input1_data, input1_dims), in TestSubFloat()
118 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubQuantized() local
127 input1_dims, input1_scale, in TestSubQuantized()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/
Dmemory_helpers_test.cc197 int input1_dims[] = {1, 1}; in TF_LITE_MICRO_TEST() local
201 nullptr, tflite::testing::IntArrayFromInts(input1_dims)); in TF_LITE_MICRO_TEST()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/
Dcommon.h774 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast() argument
784 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()
785 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()
793 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast()