Searched defs:tpu_core (Results 1 – 2 of 2) sorted by relevance
221 void SetOpSharding(Operation* op, int64_t tpu_core) { in SetOpSharding()263 const Optional<int64_t>& tpu_core, Value token) { in CreateSendOp()284 const Optional<int64_t>& tpu_core, Value token) { in CreateRecvOp()338 int64_t tpu_core = host_compute.tpu_coreAttr().getInt(); in RewriteHostComputeOp() local
315 int tpu_core; in TEST_F() local