Home
last modified time | relevance | path

Searched refs:CreateOrGetXlaSplitNodeForDistributedArg (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc1033 xla::StatusOr<ShardedInputInfo> CreateOrGetXlaSplitNodeForDistributedArg( in CreateOrGetXlaSplitNodeForDistributedArg() function
3574 CreateOrGetXlaSplitNodeForDistributedArg( in BuildExecuteNodes()