Home
last modified time | relevance | path

Searched defs:input2_dims (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dmaximum_minimum_test.cc33 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinFloat() local
74 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinQuantized() local
113 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMaxMinQuantizedInt32() local
Dpack_test.cc56 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsFloat() local
88 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackThreeInputsFloat() local
123 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsQuantized() local
160 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestPackTwoInputsQuantized32() local
Dconcatenation_test.cc35 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestConcatenateTwoInputs() local
78 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestConcatenateQuantizedTwoInputs() local
Dmul_test.cc75 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMulFloat() local
102 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestMulQuantized() local
Dlogical_test.cc33 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestLogicalOp() local
Dcomparisons_test.cc59 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonFloat() local
77 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonBool() local
95 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonInt() local
116 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonQuantizedUInt8() local
139 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestComparisonQuantizedInt8() local
Dadd_test.cc95 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestAddFloat() local
122 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestAddQuantized() local
Dsub_test.cc94 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestSubFloat() local
121 TfLiteIntArray* input2_dims = IntArrayFromInts(input2_dims_data); in TestSubQuantized() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h623 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
636 const T* input2_data, const Dims<4>& input2_dims, in Div()
980 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
993 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
1017 const Dims<4>& input2_dims, bool* output_data, in BroadcastComparison()
1032 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastComparison()
1259 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
1296 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1313 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1352 const float* input2_data, const Dims<4>& input2_dims, in Add()
[all …]
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h216 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestQS8() local
354 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF16() local
505 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF32() local
/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers_test.cc198 const int input2_dims[] = {kDimsLen, 5, 5, 5, 5}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3437 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
3494 const float* input2_data, const Dims<4>& input2_dims, in Add()
3511 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
3548 const int32* input2_data, const Dims<4>& input2_dims, in Add()
3563 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3578 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3620 const Dims<4>& input2_dims, int32 input2_offset, int32 input2_multiplier, in BroadcastAddFivefold()
3663 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3676 const Dims<4>& input2_dims, int input2_shift, in Add()
3702 const float* input2_data, const Dims<4>& input2_dims, in Sub()
[all …]