Home
last modified time | relevance | path

Searched refs:TFE_Py_TensorShapeSlice (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dtensor_test.py505 r = pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2, t3], 0)
508 r = pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2, t3], 1)
512 a = pywrap_tfe.TFE_Py_TensorShapeSlice([], 0)
522 pywrap_tfe.TFE_Py_TensorShapeSlice([t1, "abc"], 0)
527 pywrap_tfe.TFE_Py_TensorShapeSlice([2, t1], 0)
535 pywrap_tfe.TFE_Py_TensorShapeSlice(t1, -2)
542 pywrap_tfe.TFE_Py_TensorShapeSlice([t1], -2)
562 pywrap_tfe.TFE_Py_TensorShapeSlice([t1], 2)
568 pywrap_tfe.TFE_Py_TensorShapeSlice([t2], 1)
574 pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2], 1)
[all …]
H A Dpywrap_tfe.h366 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim);
H A Dpywrap_tensor.cc1122 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice() function
/aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/
H A Dsymbols_pybind.txt200 TFE_Py_TensorShapeSlice
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Darray_grad.py117 sizes = pywrap_tfe.TFE_Py_TensorShapeSlice(input_values,
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc1203 TFE_Py_TensorShapeSlice(tensors.ptr(), slice_dim)); in PYBIND11_MODULE()