Searched refs:construct_sharded_pipe (Results 1 – 1 of 1) sorted by relevance
3368 def construct_sharded_pipe(): function3383 dp, sharding_pipes = construct_sharded_pipe()3397 dp, _ = construct_sharded_pipe()3402 dp, _ = construct_sharded_pipe()3419 def construct_sharded_pipe(): function3434 dp, sharding_pipes = construct_sharded_pipe()3450 dp, _ = construct_sharded_pipe()3457 dp, _ = construct_sharded_pipe()