Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.h197 const RunTimeOperandInfo* cell_bias, const RunTimeOperandInfo* output_gate_bias,
DLSTM.cpp144 const RunTimeOperandInfo* forget_gate_bias, const RunTimeOperandInfo* cell_bias, in CheckInputTensorDimensions() argument
238 NN_CHECK_EQ(NumDimensions(cell_bias), 1); in CheckInputTensorDimensions()
239 NN_CHECK_EQ(SizeOfDimension(cell_bias, 0), n_cell); in CheckInputTensorDimensions()