/external/tensorflow/tensorflow/python/lib/core/ |
ndarray_tensor.cc | 443 *out_tensor = make_safe(TF_NewTensor(dtype, {}, 0, PyArray_DATA(array), 449 *out_tensor = make_safe(TF_NewTensor(dtype, dims.data(), dims.size(), 457 make_safe(TF_NewTensor(dtype, dims.data(), dims.size(), encoded, size,
|
/external/tensorflow/tensorflow/c/ |
c_test_util.cc | 49 return TF_NewTensor(TF_BOOL, nullptr, 0, values, num_bytes, &BoolDeallocator, 85 return TF_NewTensor(TF_INT32, nullptr, 0, values, num_bytes, 93 return TF_NewTensor(TF_DOUBLE, nullptr, 0, values, num_bytes, 101 return TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes,
|
c_api.cc | 200 return TF_NewTensor(dtype, dims, num_dims, data, len, deallocate_buffer, 204 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, 225 buf = new TF_ManagedBuffer(allocate_tensor("TF_NewTensor", len), len, 556 return TF_NewTensor( 643 return TF_NewTensor(TF_STRING, [all...] |
c_api.h | 234 TF_CAPI_EXPORT extern TF_Tensor* TF_NewTensor( 241 // This function is an alternative to TF_NewTensor and should be used when [all...] |
c_api_test.cc | 133 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, 153 TF_NewTensor(TF_FLOAT, nullptr, 0, nullptr, 0, &NoOpDeallocator, nullptr); 177 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, [all...] |