Searched defs:shape_c (Results 1 – 3 of 3) sorted by relevance
28 int64_t shape_c = 1; member
239 ShapeVector shape_c(void) const { return shape(); } in shape_c() function
268 std::vector<int64_t> shape_c{1, kUnidirectionalGateNum * hidden_size, hidden_size}; in SplitWeights() local