Searched refs:AddLayerNormCoeffsTensor (Results 1 – 2 of 2) sorted by relevance
113 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
3252 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