Searched refs:ShardedInputInfo (Results 1 – 1 of 1) sorted by relevance
651 struct ShardedInputInfo { struct791 xla::StatusOr<ShardedInputInfo> CreateOrGetSplitNodesForInputSharding( in CreateOrGetSplitNodesForInputSharding()796 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetSplitNodesForInputSharding()898 ShardedInputInfo sharded_input_info{root_split_node, in CreateOrGetSplitNodesForInputSharding()3054 std::map<ShardedInputIndex, ShardedInputInfo> input_index_to_sharded_inputs; in BuildExecuteNodes()3170 ShardedInputInfo sharded_input_info, in BuildExecuteNodes()3253 ShardedInputInfo sharded_input_info, in BuildExecuteNodes()