Searched refs:kUnidirectionalGateNum (Results 1 – 2 of 2) sorted by relevance
38 constexpr auto kUnidirectionalGateNum = 4; variable262 std::vector<int64_t> shape_i{1, kUnidirectionalGateNum * hidden_size, input_size}; in SplitWeights()268 std::vector<int64_t> shape_c{1, kUnidirectionalGateNum * hidden_size, hidden_size}; in SplitWeights()
40 constexpr auto kUnidirectionalGateNum = 4; variable476 …new_shape = std::vector<int64_t>({1, data_shapes[0][0] * kUnidirectionalGateNum, data_shapes[0][1]… in GetConcatedParam()