Searched refs:cell_bias (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup_test.cc | 260 std::vector<int32_t> cell_bias; in TEST() local
263 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
283 std::vector<int32_t> cell_bias; in TEST() local
286 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
289 EXPECT_THAT(cell_bias, ElementsAreArray({39481, 48624, 48976, -21419})); in TEST()
306 std::vector<int32_t> cell_bias; in TEST() local
309 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
329 std::vector<int32_t> cell_bias; in TEST() local
332 DecomposeBiasTensor(biases.data(), 4, &input_bias, &cell_bias, &forget_bias, in TEST()
|
| D | quant_lstm_sup.cc | 132 std::vector<int32_t>* cell_bias, in DecomposeBiasTensor() argument
138 cell_bias->resize(bias_size); in DecomposeBiasTensor()
139 std::copy(biases + bias_size, biases + 2 * bias_size, cell_bias->begin()); in DecomposeBiasTensor()
|
| D | quant_lstm_sup.h | 50 std::vector<int32_t>* cell_bias,
|
| D | nnapi_delegate.cc | 3299 std::vector<int32_t> cell_bias; in Map() local
3303 &input_bias, &cell_bias, in Map()
3316 ANEURALNETWORKS_TENSOR_INT32, kTfLiteInt32, {bias_size}, cell_bias, in Map()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 353 const TfLiteTensor* cell_bias, const TfLiteTensor* output_gate_bias, in EvalFloat() argument
439 GetTensorData<float>(cell_bias), in EvalFloat()
501 GetTensorData<float>(cell_bias), in EvalFloat()
581 const TfLiteTensor* cell_bias = in lstm_eval() local
624 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 366 self.cell_bias = add_variable_wrapped(
486 transpose_b=True), self.cell_bias)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 169 lstm.cell_bias(), builder); in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3856 TFL_TensorOf<[F32, QI32]>:$cell_bias,
3980 TFL_FpTensor:$cell_bias,
|