Searched refs:_get_input_tensor (Results 1 – 2 of 2) sorted by relevance
123 def _get_input_tensor(self, node_def, input_index): member in NodeFileWriterTest187 self.assertIsNone(self._get_input_tensor(node_def, 0))189 self._get_input_tensor(node_def, 1), np.array([[1, 2], [3, 4]]))190 self.assertIsNone(self._get_input_tensor(node_def, 2))
4310 def _get_input_tensor(time): function4320 inp = _get_input_tensor(i)4357 inp = _get_input_tensor(i)