Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dtensor_test.py448 r = pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2, t3], 0)
451 r = pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2, t3], 1)
455 a = pywrap_tfe.TFE_Py_TensorShapeSlice([], 0)
465 pywrap_tfe.TFE_Py_TensorShapeSlice([t1, "abc"], 0)
470 pywrap_tfe.TFE_Py_TensorShapeSlice([2, t1], 0)
478 pywrap_tfe.TFE_Py_TensorShapeSlice(t1, -2)
485 pywrap_tfe.TFE_Py_TensorShapeSlice([t1], -2)
505 pywrap_tfe.TFE_Py_TensorShapeSlice([t1], 2)
511 pywrap_tfe.TFE_Py_TensorShapeSlice([t2], 1)
517 pywrap_tfe.TFE_Py_TensorShapeSlice([t1, t2], 1)
[all …]
Dpywrap_tfe.h362 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim);
Dpywrap_tensor.cc1016 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice() function
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt191 TFE_Py_TensorShapeSlice
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py120 sizes = pywrap_tfe.TFE_Py_TensorShapeSlice(input_values,
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc1033 TFE_Py_TensorShapeSlice(tensors.ptr(), slice_dim)); in PYBIND11_MODULE()