Searched refs:set_configuration_from_sharded_input_tensors (Results 1 – 2 of 2) sorted by relevance
88 i.set_configuration_from_sharded_input_tensors([[t2, t1], [t2, t1]])102 i.set_configuration_from_sharded_input_tensors([[t2, t1], [t2, t1]])
396 def set_configuration_from_sharded_input_tensors(self, input_tensors): member in InfeedQueue616 self.set_configuration_from_sharded_input_tensors(sharded_inputs)855 self.set_configuration_from_sharded_input_tensors(sharded_inputs)