Searched defs:sharded_input_info (Results 1 – 1 of 1) sorted by relevance
910 ShardedInputInfo sharded_input_info{root_split_node, in CreateOrGetSplitNodesForInputSharding() local1027 ShardedInputInfo sharded_input_info{nullptr, std::move(sharded_inputs_list)}; in CreateOrGetXlaSplitNodeForShardedPerReplicaArg() local1054 ShardedInputInfo sharded_input_info{nullptr, std::move(sharded_inputs_list)}; in CreateOrGetXlaSplitNodeForDistributedArg() local1114 ShardedInputInfo sharded_input_info{nullptr, std::move(sharded_inputs_list)}; in CreateOrGetXlaSplitNodeForVariableArg() local3121 ShardedInputInfo sharded_input_info{nullptr, in CreatePartitionedDummyVarArgs() local3563 ShardedInputInfo sharded_input_info; in BuildExecuteNodes() local3664 ShardedInputInfo sharded_input_info; in BuildExecuteNodes() local