| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mul.h | 25 const int8_t* input1_data, const int8_t* input2_data, in MulElementwise() argument
29 const int32 input2_val = params.input2_offset + input2_data[i]; in MulElementwise()
44 const RuntimeShape& input2_shape, const int8_t* input2_data, in Mul() argument
52 MulElementwise(flat_size, params, input1_data, input2_data, output_data); in Mul()
58 const RuntimeShape& input2_shape, const int16* input2_data, in Mul() argument
74 F0::FromRaw(input1_data[i]) * F0::FromRaw(input2_data[i]); in Mul()
89 const int8_t* input2_data, in BroadcastMul4DSlow() argument
111 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastMul4DSlow()
|
| D | add.h | 29 const int8_t* input1_data, const int8_t* input2_data, in AddElementwise() argument
39 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
62 const RuntimeShape& input2_shape, const int8_t* input2_data, in Add() argument
74 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
81 const int8_t* input2_data, in BroadcastAdd4DSlow() argument
112 input2_data[SubscriptToIndex(desc2, b, y, x, c)]; in BroadcastAdd4DSlow()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 573 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
580 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
586 const T* input2_data, const Dims<4>& input2_dims, in Div() argument
593 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Div()
889 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
894 DimsToShape(input2_dims), input2_data, in Comparison()
902 const T* input2_data, const Dims<4>& input2_dims, in Comparison() argument
918 DimsToShape(input2_dims), input2_data, in Comparison()
925 const T* input2_data, in BroadcastComparison() argument
932 input2_data, DimsToShape(output_dims), in BroadcastComparison()
[all …]
|
| D | reference_ops.h | 527 const RuntimeShape& input2_shape, const T* input2_data, in Add() argument
533 input1_data[i] + input2_data[i], params.quantized_activation_min, in Add()
540 const RuntimeShape& input2_shape, const float* input2_data, in Add() argument
544 auto x = input1_data[i] + input2_data[i]; in Add()
569 const uint8* input1_data, const uint8* input2_data, in AddElementwise() argument
578 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
603 uint8 input1_data, const uint8* input2_data, in AddScalarBroadcast() argument
616 const int32 input2_val = params.input2_offset + input2_data[i]; in AddScalarBroadcast()
635 const RuntimeShape& input2_shape, const uint8* input2_data, in Add() argument
646 AddElementwise(flat_size, params, input1_data, input2_data, output_data); in Add()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 874 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv() argument
881 DimsToShape(input2_dims), input2_data, in BroadcastDiv()
931 const float* input2_data, const Dims<4>& input2_dims, in Add() argument
940 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
948 const uint8* input2_data, const Dims<4>& input2_dims, in Add() argument
979 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
985 const int32* input2_data, const Dims<4>& input2_dims, in Add() argument
994 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims), in Add()
1000 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd() argument
1007 DimsToShape(input2_dims), input2_data, in BroadcastAdd()
[all …]
|
| D | optimized_ops.h | 2625 const RuntimeShape& input2_shape, const float* input2_data, in Add() argument
2639 auto a20 = vld1q_f32(input2_data + i); in Add()
2640 auto a21 = vld1q_f32(input2_data + i + 4); in Add()
2641 auto a22 = vld1q_f32(input2_data + i + 8); in Add()
2642 auto a23 = vld1q_f32(input2_data + i + 12); in Add()
2662 auto a2 = vld1q_f32(input2_data + i); in Add()
2671 auto x = input1_data[i] + input2_data[i]; in Add()
2680 const uint8* input1_data, const uint8* input2_data, in AddElementwise() argument
2695 const uint8x8_t input2_val_original = vld1_u8(input2_data + i); in AddElementwise()
2747 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2842 const RuntimeShape& input2_shape, const float* input2_data, in Add() argument
2856 auto a20 = vld1q_f32(input2_data + i); in Add()
2857 auto a21 = vld1q_f32(input2_data + i + 4); in Add()
2858 auto a22 = vld1q_f32(input2_data + i + 8); in Add()
2859 auto a23 = vld1q_f32(input2_data + i + 12); in Add()
2879 auto a2 = vld1q_f32(input2_data + i); in Add()
2888 auto x = input1_data[i] + input2_data[i]; in Add()
2897 const uint8* input1_data, const uint8* input2_data, in AddElementwise() argument
2911 const auto input2_val_original = vld1_u8(input2_data + i); in AddElementwise()
2963 const int32 input2_val = params.input2_offset + input2_data[i]; in AddElementwise()
[all …]
|