Searched defs:ShardedInputIndex (Results 1 – 1 of 1) sorted by relevance
644 struct ShardedInputIndex { struct645 int replica_id;646 int argument_index;648 bool operator<(const ShardedInputIndex& rhs) const { in operator <()807 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetSplitNodesForInputSharding()1011 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()1038 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForDistributedArg()1069 std::map<ShardedInputIndex, ShardedInputInfo>* in CreateOrGetXlaSplitNodeForVariableArg()3081 std::map<ShardedInputIndex, ShardedInputInfo>* in CreatePartitionedDummyVarArgs()