| 169 | int TF_NumDims(const TF_Tensor* t) { return t->tensor.dims(); } |
| 170 | |
| 171 | int64_t TF_Dim(const TF_Tensor* t, int dim_index) { |
| 172 | return static_cast<int64_t>(t->tensor.dim_size(dim_index)); |
| 173 | } |
| 174 | |
| 175 | size_t TF_TensorByteSize(const TF_Tensor* t) { |
| 176 | return tensorflow::TensorCApi::Buffer(t->tensor)->size(); |