MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Sub

Function Sub

tensorflow/lite/kernels/internal/optimized/legacy_optimized_ops.h:3686–3698  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3684}
3685
3686inline void Sub(const float* input1_data, const Dims<4>& input1_dims,
3687 const float* input2_data, const Dims<4>& input2_dims,
3688 float* output_data, const Dims<4>& output_dims) {
3689 float output_activation_min, output_activation_max;
3690 GetActivationMinMax(FusedActivationFunctionType::kNone,
3691 &output_activation_min, &output_activation_max);
3692 tflite::ArithmeticParams op_params;
3693 op_params.float_activation_min = output_activation_min;
3694 op_params.float_activation_max = output_activation_max;
3695 Sub(op_params, DimsToShape(input1_dims), input1_data,
3696 DimsToShape(input2_dims), input2_data, DimsToShape(output_dims),
3697 output_data);
3698}
3699
3700template <typename T>
3701void Sub(const T* input1_data, const Dims<4>& input1_dims, const T* input2_data,

Callers

nothing calls this directly

Calls 2

GetActivationMinMaxFunction · 0.85
DimsToShapeFunction · 0.85

Tested by

no test coverage detected