Searched refs:_CreateNumpyTensor (Results 1 – 2 of 2) sorted by relevance
176 def _CreateNumpyTensor(self, shape): member in Conv2DTest200 x1 = self._CreateNumpyTensor(tensor_in_sizes)201 x2 = self._CreateNumpyTensor(filter_in_sizes)268 x1 = self._CreateNumpyTensor(tensor_in_sizes)269 x2 = self._CreateNumpyTensor(filter_in_sizes)400 input_tensor = self._CreateNumpyTensor(tensor_in_sizes)401 filter_tensor = self._CreateNumpyTensor(filter_in_sizes)439 filter_in = self._CreateNumpyTensor(filter_in_sizes)440 x1 = self._CreateNumpyTensor(tensor_in_sizes_batch)463 filter_in = self._CreateNumpyTensor(filter_in_sizes)[all …]
192 def _CreateNumpyTensor(self, sizes): member in Conv3DTest202 filter_in = self._CreateNumpyTensor(filter_in_sizes)203 x1 = self._CreateNumpyTensor(tensor_in_sizes_batch)218 filter_in = self._CreateNumpyTensor(filter_in_sizes)219 x1 = self._CreateNumpyTensor(tensor_in_sizes_batch)244 filter_in = self._CreateNumpyTensor(filter_in_sizes)245 x1 = self._CreateNumpyTensor(tensor_in_sizes_batch)462 x1 = self._CreateNumpyTensor(tensor_in_sizes)