Searched refs:bias_vec (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 234 auto* bias_vec = bias->mutable_vector(); in Compute() local
237 bias_vec->set_value(idx, bias_vec->value(idx) + delta); in Compute()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 584 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCell() local
586 reduced = vaddq_s32(reduced, bias_vec); in GEMVForLstmCell()
637 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCellWithSymmetricRange() local
880 reduced = vaddq_s32(reduced, bias_vec); in GEMVForLstmCellWithSymmetricRange()
1089 int32x4_t bias_vec = vld1q_s32(bias_ptr); in FullyConnectedAsGEMV() local
1091 reduced = vaddq_s32(reduced, bias_vec); in FullyConnectedAsGEMV()
1435 int32x4_t bias_vec = vld1q_s32(bias_data + c); in ShuffledFullyConnectedWorkerImpl() local
1436 reduced = vaddq_s32(reduced, bias_vec); in ShuffledFullyConnectedWorkerImpl()
1526 int32x4_t bias_vec = vld1q_s32(bias_data + c); \ in ShuffledFullyConnectedWorkerImpl()
1527 reduced = vaddq_s32(reduced, bias_vec); \ in ShuffledFullyConnectedWorkerImpl()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 462 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCell() local
464 reduced = vaddq_s32(reduced, bias_vec); in GEMVForLstmCell()
515 int32x4_t bias_vec = vld1q_s32(bias_ptr); in GEMVForLstmCellWithSymmetricRange() local
758 reduced = vaddq_s32(reduced, bias_vec); in GEMVForLstmCellWithSymmetricRange()
1002 int32x4_t bias_vec = vld1q_s32(bias_data + out); in FullyConnectedAsGEMVWorkerImpl() local
1003 reduced = vaddq_s32(reduced, bias_vec); in FullyConnectedAsGEMVWorkerImpl()
1379 int32x4_t bias_vec = vld1q_s32(bias_data + c); in ShuffledFullyConnectedWorkerImpl() local
1380 reduced = vaddq_s32(reduced, bias_vec); in ShuffledFullyConnectedWorkerImpl()
1470 int32x4_t bias_vec = vld1q_s32(bias_data + c); \ in ShuffledFullyConnectedWorkerImpl()
1471 reduced = vaddq_s32(reduced, bias_vec); \ in ShuffledFullyConnectedWorkerImpl()
|