Searched defs:arg_index_to_sharded_input_map (Results 1 – 1 of 1) sorted by relevance
808 arg_index_to_sharded_input_map) { in CreateOrGetSplitNodesForInputSharding()1012 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForShardedPerReplicaArg()1039 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForDistributedArg()1070 arg_index_to_sharded_input_map) { in CreateOrGetXlaSplitNodeForVariableArg()3082 arg_index_to_sharded_input_map) { in CreatePartitionedDummyVarArgs()