Searched refs:input_diff (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 56 int32 input_diff = in Softmax() local
58 if (input_diff >= diff_min) { in Softmax()
61 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
74 int32 input_diff = in Softmax() local
76 if (input_diff >= diff_min) { in Softmax()
79 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|
| D | log_softmax.h | 53 int32_t input_diff = in LogSoftmax() local
56 if (input_diff >= diff_min) { in LogSoftmax()
58 input_diff, input_multiplier, input_shift); in LogSoftmax()
81 int32_t input_diff = in LogSoftmax() local
85 if (input_diff > adjusted_diff_min) { in LogSoftmax()
87 input_diff, input_multiplier, input_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 92 int32 input_diff = in Softmax() local
94 if (input_diff >= diff_min) { in Softmax()
97 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
110 int32 input_diff = in Softmax() local
112 if (input_diff >= diff_min) { in Softmax()
115 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|
| D | reference_ops.h | 2634 int32 input_diff = in LogSoftmax() local
2636 if (input_diff >= diff_min) { in LogSoftmax()
2639 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
2666 int32 input_diff = in LogSoftmax() local
2668 if (input_diff > adjusted_diff_min) { in LogSoftmax()
2671 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4573 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4574 if (input_diff >= diff_min) { in Softmax()
4577 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4640 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4641 if (input_diff >= diff_min) { in Softmax()
4644 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4737 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
4738 if (input_diff >= diff_min) { in LogSoftmax()
4741 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
4768 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4698 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4699 if (input_diff >= diff_min) { in Softmax()
4702 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4765 int32 input_diff = static_cast<int32>(input_data_ptr[c]) - max_in_row; in Softmax() local
4766 if (input_diff >= diff_min) { in Softmax()
4769 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4998 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
4999 if (input_diff >= diff_min) { in LogSoftmax()
5002 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
5029 int32 input_diff = static_cast<int32>(block_input_data[c]) - max_in_row; in LogSoftmax() local
[all …]
|