Searched refs:_read_tpu_coreid_from_infeed (Results 1 – 1 of 1) sorted by relevance
899 def _read_tpu_coreid_from_infeed(mode, infeed_tensors): function965 _read_tpu_coreid_from_infeed(