Searched refs:infeed_tensors (Results 1 – 1 of 1) sorted by relevance
639 for infeed_tensors, inputs in zip(tpu_model_op.infeed_tensors,641 for tensor, value in zip(infeed_tensors, inputs):686 infeed_tensors = []690 infeed_tensors.append(695 shard_infeed_tensors.append(infeed_tensors)699 infeed_tensors, [spec.shape for spec in input_specs],899 def _read_tpu_coreid_from_infeed(mode, infeed_tensors): argument902 return None, infeed_tensors904 if len(infeed_tensors) <= 1:908 len(infeed_tensors), infeed_tensors))[all …]