Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test_util.h113 AddLayerNormCoeffsTensor(20, input_shapes); in n_batch_()
116 AddLayerNormCoeffsTensor(21, input_shapes); in n_batch_()
118 AddLayerNormCoeffsTensor(22, input_shapes); in n_batch_()
120 AddLayerNormCoeffsTensor(23, input_shapes); in n_batch_()
272 int AddLayerNormCoeffsTensor( in AddLayerNormCoeffsTensor() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3252 AddLayerNormCoeffsTensor(kInputLayerNormCoeffsIndex, input_shapes); in LSTMOpModel()
3255 AddLayerNormCoeffsTensor(kForgetLayerNormCoeffsIndex, input_shapes); in LSTMOpModel()
3257 AddLayerNormCoeffsTensor(kCellLayerNormCoeffsIndex, input_shapes); in LSTMOpModel()
3259 AddLayerNormCoeffsTensor(kOutputLayerNormCoeffsIndex, input_shapes); in LSTMOpModel()
3410 int AddLayerNormCoeffsTensor( in AddLayerNormCoeffsTensor() function in tflite::__anon1a74836d0111::LSTMOpModel