Searched refs:TFE_TensorHandleDim (Results 1 – 3 of 3) sorted by relevance
355 PyObject* dim = PyLong_FromLongLong(TFE_TensorHandleDim(handle, i)); in EagerTensor_shape_tuple()700 int64_t dim = TFE_TensorHandleDim(handle, slice_dim); in TFE_Py_TensorShapeSlice()
123 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h,
159 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index) { in TFE_TensorHandleDim() function