Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dlegacy_reference_ops.h625 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
638 const T* input2_data, const Dims<4>& input2_dims, in Div()
983 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
996 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
1020 const Dims<4>& input2_dims, bool* output_data, in BroadcastComparison()
1035 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastComparison()
1262 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
1299 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1316 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1355 const float* input2_data, const Dims<4>& input2_dims, in Add()
[all …]
/aosp_15_r20/external/XNNPACK/test/
H A Dbinary-elementwise-operator-tester.h215 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestQS8() local
397 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestQU8() local
572 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF16() local
793 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; in TestF32() local
H A Dsubgraph-binary-tester.h113 std::array<size_t, XNN_MAX_TENSOR_DIMS> input2_dims; variable
H A Dconcatenate2.cc104 std::vector<size_t> input2_dims; member in Concatenate2Test
H A Dconcatenate3.cc109 std::vector<size_t> input2_dims; member in Concatenate3Test
H A Dconcatenate4.cc114 std::vector<size_t> input2_dims; member in Concatenate4Test
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Dlegacy_optimized_ops.h3438 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
3495 const float* input2_data, const Dims<4>& input2_dims, in Add()
3512 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
3549 const int32* input2_data, const Dims<4>& input2_dims, in Add()
3564 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3579 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3621 const Dims<4>& input2_dims, int32 input2_offset, int32 input2_multiplier, in BroadcastAddFivefold()
3664 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
3677 const Dims<4>& input2_dims, int input2_shift, in Add()
3703 const float* input2_data, const Dims<4>& input2_dims, in Sub()
[all …]