Searched defs:ShardedInputIndex (Results 1 – 1 of 1) sorted by relevance
641 struct ShardedInputIndex { struct642 int replica_id;643 int argument_index;645 bool operator<(const ShardedInputIndex& rhs) const { in operator <()809 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetSplitNodesForInputSharding()1015 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()1042 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForDistributedArg()1073 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForVariableArg()3139 std::map<ShardedInputIndex, ShardedInputInfo>* in CreatePartitionedDummyVarArgs()