Searched refs:TFE_TensorHandleDim (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/go/ |
| D | tensor_handle.go | 101 n := int64(C.TFE_TensorHandleDim(th.c, C.int(index), status.c))
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 40 shape.push_back(TFE_TensorHandleDim(handle, i, status)); in TensorShapeAsVector()
|
| D | c_api.h | 170 TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h,
|
| D | c_api.cc | 466 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim() function
|
| D | c_api_test.cc | 810 int dim = TFE_TensorHandleDim(h, 0, status.get()); in TEST()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 549 PyLong_FromLongLong(TFE_TensorHandleDim(handle, i, self->status)); in EagerTensor_shape_tuple()
1031 int64_t dim = TFE_TensorHandleDim(handle, slice_dim, status.get()); in TFE_Py_TensorShapeSlice()
|