| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | legacy_reference_ops.h | 624 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
637 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div()
982 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison()
994 const Dims<4>& input1_dims, int32 input1_offset, in Comparison()
1018 const Dims<4>& input1_dims, in BroadcastComparison()
1032 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastComparison()
1260 const Dims<4>& input1_dims, int32 input1_offset, in Add()
1298 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
1314 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
1354 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
[all …]
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | binary-elementwise-operator-tester.h | 214 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestQS8() local
396 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestQU8() local
571 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF16() local
792 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; in TestF32() local
|
| H A D | subgraph-binary-tester.h | 112 std::array<size_t, XNN_MAX_TENSOR_DIMS> input1_dims; variable
|
| H A D | concatenate2.cc | 103 std::vector<size_t> input1_dims; member in Concatenate2Test
|
| H A D | concatenate3.cc | 108 std::vector<size_t> input1_dims; member in Concatenate3Test
|
| H A D | concatenate4.cc | 113 std::vector<size_t> input1_dims; member in Concatenate4Test
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | legacy_optimized_ops.h | 3437 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
3494 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
3510 const Dims<4>& input1_dims, int32 input1_offset, in Add()
3548 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
3563 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
3577 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
3619 const uint8* input1_data, const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAddFivefold()
3663 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
3675 inline void Add(const int16* input1_data, const Dims<4>& input1_dims, in Add()
3702 inline void Sub(const float* input1_data, const Dims<4>& input1_dims, in Sub()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | add_n.cc | 98 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | common.h | 915 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast()
|