Searched defs:tensor_class (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_reshape_op_test.py | 354 def _MakeAndReshapeTensor(self, tensor_class, original_shape, target_shape): argument
372 def testImpliedReshapeEmpty1DTensor(self, tensor_class): argument
382 def testImpliedReshapeEmpty2DTensor(self, tensor_class): argument
392 def testImpliedReshapeEmpty3DTensor(self, tensor_class): argument
401 def testImpliedReshapeEmpty4DTensor(self, tensor_class): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | keras_tensor.py | 362 def _overload_all_operators(cls, tensor_class): # pylint: disable=invalid-name argument
373 def _overload_operator(cls, tensor_class, operator): # pylint: disable=invalid-name argument
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen_lazy_tensor.py | 176 tensor_class: str = "torch::lazy::LazyTensor" variable in default_args
|
| /aosp_15_r20/external/pytorch/torch/csrc/tensor/ |
| H A D | python_tensor.cpp | 265 auto tensor_class = THPObjectPtr(PyObject_GetAttrString(torch, "Tensor")); in get_tensor_dict() local
|