Searched defs:gate_shape (Results 1 – 1 of 1) sorted by relevance
490 const std::vector<int64_t> gate_shape{input_size + hidden_size, hidden_size * kGateNum}; in ConvertWeightData() local542 std::vector<int64_t> gate_shape{hidden_size * kGateNum}; in ConvertBiasData() local