Searched defs:split_sizes (Results 1 – 7 of 7) sorted by relevance
35 auto split_sizes = split_prim_->size_splits(); in BuildLayer() local
56 auto split_sizes = in ChunkAscendCustomize() local
27 const int *split_sizes = split_param->split_sizes_; in DoSplit() local
30 const int32_t *split_sizes = param->split_sizes_; in Int8DoSplit() local
85 auto split_sizes = std::make_unique<int[]>(IntToSize(param.num_split_)); in LaunchSplit() local
40 std::vector<int> split_sizes = {4, 15, 11}; in TEST_F() local
3326 std::vector<int64_t> split_sizes = {input_size * gate_size, hidden_size * gate_size}; in ExportPrimDynamicRNN() local3450 std::vector<int64_t> split_sizes = {input_size * gate_size, hidden_size * gate_size}; in ExportLSTMWeights() local