Home
last modified time | relevance | path

Searched refs:input_bias (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dquant_lstm_sup_test.cc259 std::vector<int32_t> input_bias; in TEST() local
263 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
266 EXPECT_THAT(input_bias, ElementsAreArray({-7876, 13488, -726, 32839})); in TEST()
282 std::vector<int32_t> input_bias; in TEST() local
286 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
305 std::vector<int32_t> input_bias; in TEST() local
309 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
328 std::vector<int32_t> input_bias; in TEST() local
332 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
Dquant_lstm_sup.cc131 std::vector<int32_t>* input_bias, in DecomposeBiasTensor() argument
135 input_bias->resize(bias_size); in DecomposeBiasTensor()
136 std::copy(biases, biases + bias_size, input_bias->begin()); in DecomposeBiasTensor()
Dquant_lstm_sup.h49 std::vector<int32_t>* input_bias,
Dnnapi_delegate.cc3298 std::vector<int32_t> input_bias; in Map() local
3303 &input_bias, &cell_bias, in Map()
3308 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, input_bias, in Map()
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn_cell.py361 self.input_bias = add_variable_wrapped(
468 transpose_b=True), self.input_bias)
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent.py1839 input_bias, recurrent_bias = self.bias, None
1841 input_bias, recurrent_bias = array_ops.unstack(self.bias)
1858 x_z = K.bias_add(x_z, input_bias[:self.units])
1859 x_r = K.bias_add(x_r, input_bias[self.units: self.units * 2])
1860 x_h = K.bias_add(x_h, input_bias[self.units * 2:])
1901 matrix_x = K.bias_add(matrix_x, input_bias)
Drecurrent_v2.py587 input_bias, recurrent_bias = array_ops.unstack(bias)
595 matrix_x = K.bias_add(matrix_x, input_bias)