Searched defs:tpu_metadata (Results 1 – 1 of 1) sorted by relevance
1229 TPUMetadata tpu_metadata; in ComputeAsync() local1276 TPUMetadata tpu_metadata; in ComputeAsync() local1636 bool enable_spmd_xla_partitioning, const TPUMetadata& tpu_metadata) { in ReplaceResourceArgsWithVarHandleOps()1799 ResourceHandle& handle, Node* variable, const TPUMetadata& tpu_metadata) { in ReplaceAndPartitionXLAShardingVariable()2293 TPUMetadata* tpu_metadata) { in GetGraphFromFunction()