Searched defs:arg_index_to_sharded_input_map (Results 1 – 1 of 1) sorted by relevance
810 arg_index_to_sharded_input_map) { in CreateOrGetSplitNodesForInputSharding()1016 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()1043 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForDistributedArg()1074 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForVariableArg()3140 arg_index_to_sharded_input_map) { in CreatePartitionedDummyVarArgs()