Home
last modified time | relevance | path

Searched defs:input1_dims (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h622 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
635 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div()
979 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison()
991 const Dims<4>& input1_dims, int32 input1_offset, in Comparison()
1015 const Dims<4>& input1_dims, in BroadcastComparison()
1029 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastComparison()
1257 const Dims<4>& input1_dims, int32 input1_offset, in Add()
1295 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
1311 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
1351 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dmaximum_minimum_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinFloat() local
73 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantized() local
112 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMaxMinQuantizedInt32() local
Dpack_test.cc55 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsFloat() local
87 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackThreeInputsFloat() local
122 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized() local
159 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestPackTwoInputsQuantized32() local
Dconcatenation_test.cc34 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateTwoInputs() local
77 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestConcatenateQuantizedTwoInputs() local
Dadd_n.cc54 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
Dmul_test.cc74 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulFloat() local
101 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestMulQuantized() local
Dlogical_test.cc32 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestLogicalOp() local
Dcomparisons_test.cc58 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonFloat() local
76 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonBool() local
94 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonInt() local
115 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedUInt8() local
138 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestComparisonQuantizedInt8() local
Dadd_test.cc94 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddFloat() local
121 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestAddQuantized() local
Dsub_test.cc93 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubFloat() local
120 TfLiteIntArray* input1_dims = IntArrayFromInts(input1_dims_data); in TestSubQuantized() local
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc54 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h215 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestQS8() local
353 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF16() local
504 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF32() local
/external/tensorflow/tensorflow/lite/micro/
Dmemory_helpers_test.cc197 const int input1_dims[] = {1, 1}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h3436 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
3493 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
3509 const Dims<4>& input1_dims, int32 input1_offset, in Add()
3547 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
3562 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
3576 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
3618 const uint8* input1_data, const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAddFivefold()
3662 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
3674 inline void Add(const int16* input1_data, const Dims<4>& input1_dims, in Add()
3701 inline void Sub(const float* input1_data, const Dims<4>& input1_dims, in Sub()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h772 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast()