Searched refs:tensor_class (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_reshape_op_test.py | 336 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
337 if tensor_class == "sparse":
354 def testImpliedReshapeEmpty1DTensor(self, tensor_class): argument
356 self._MakeAndReshapeTensor(tensor_class, [0], [-1, 1]), [0, 1])
358 self._MakeAndReshapeTensor(tensor_class, [0], [-1, 1, 2]), [0, 1, 2])
364 def testImpliedReshapeEmpty2DTensor(self, tensor_class): argument
366 self._MakeAndReshapeTensor(tensor_class, [1, 0], [-1, 1]), [0, 1])
368 self._MakeAndReshapeTensor(tensor_class, [1, 0], [-1, 2, 3]), [0, 2, 3])
374 def testImpliedReshapeEmpty3DTensor(self, tensor_class): argument
376 self._MakeAndReshapeTensor(tensor_class, [1, 0, 0], [-1, 2, 3]),
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 385 def _overload_all_operators(cls, tensor_class): # pylint: disable=invalid-name argument
388 cls._overload_operator(tensor_class, operator)
392 if hasattr(tensor_class, 'experimental_ref'):
393 cls._overload_operator(tensor_class, 'experimental_ref')
396 def _overload_operator(cls, tensor_class, operator): # pylint: disable=invalid-name argument
405 tensor_oper = getattr(tensor_class, operator)
|