Searched defs:tpu_replicate (Results 1 – 2 of 2) sorted by relevance
76 void TFOp::setTpuReplicate(StringAttr tpu_replicate) { in setTpuReplicate()
152 auto tpu_replicate = StringAttr::get(op->getContext(), "a_tpu"); in TEST() local