Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval_test.cc114 TfLiteIntArrayFree(layer_norm_cell_tensor_.dims); in ~BaseLstmParam()
273 TfLiteTensor layer_norm_cell_tensor_; member in tflite::__anonb312a86f0111::BaseLstmParam
341 PackWeightToTensor(&layer_norm_cell_tensor_, layer_norm_cell_, in GetCellLayerNorm()
343 layer_norm_cell_tensor_.data.i16 = layer_norm_cell_.data(); in GetCellLayerNorm()
344 return &layer_norm_cell_tensor_; in GetCellLayerNorm()
787 PackWeightToTensor(&layer_norm_cell_tensor_, layer_norm_float_cell_, in GetCellLayerNorm()
789 layer_norm_cell_tensor_.data.f = layer_norm_float_cell_.data(); in GetCellLayerNorm()
790 return &layer_norm_cell_tensor_; in GetCellLayerNorm()