Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dtensor_test.py375 r = pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1, t2, t3], 0)
378 r = pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1, t2, t3], 1)
382 a = pywrap_tensorflow.TFE_Py_TensorShapeSlice([], 0)
392 pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1, "abc"], 0)
397 pywrap_tensorflow.TFE_Py_TensorShapeSlice([2, t1], 0)
405 pywrap_tensorflow.TFE_Py_TensorShapeSlice(t1, -2)
413 pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1], -2)
433 pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1], 2)
439 pywrap_tensorflow.TFE_Py_TensorShapeSlice([t2], 1)
445 pywrap_tensorflow.TFE_Py_TensorShapeSlice([t1, t2], 1)
[all …]
Dpywrap_tfe.h227 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim);
Dpywrap_tensor.cc978 PyObject* TFE_Py_TensorShapeSlice(PyObject* tensors, int slice_dim) { in TFE_Py_TensorShapeSlice() function
/external/tensorflow/tensorflow/python/
Dpywrap_tfe.i77 %rename("%s") TFE_Py_TensorShapeSlice;
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py114 sizes = pywrap_tensorflow.TFE_Py_TensorShapeSlice(input_values,