| /external/tensorflow/tensorflow/python/lib/core/ |
| safe_ptr.h | 34 void operator()(TF_Tensor* p) const { TF_DeleteTensor(p); }
52 // Safe containers for an owned TF_Tensor. On destruction, the tensor will be
54 using Safe_TF_TensorPtr = std::unique_ptr<TF_Tensor, detail::TFTensorDeleter>;
55 Safe_TF_TensorPtr make_safe(TF_Tensor* tensor);
|
| safe_ptr.cc | 24 Safe_TF_TensorPtr make_safe(TF_Tensor* tensor) {
|
| ndarray_tensor.cc | 220 Status CopyTF_TensorStringsToPyArray(const TF_Tensor* src, uint64 nelems,
273 const TF_Tensor* tensor, tensorflow::int64* nelems) {
293 Status GetPyArrayDescrForTensor(const TF_Tensor* tensor,
358 // Converts the given TF_Tensor to a numpy ndarray.
373 TF_Tensor* original = tensor.get();
374 TF_Tensor* moved = TF_TensorMaybeMove(tensor.release());
403 " bytes but TF_Tensor was ",
436 // Create a TF_Tensor based on the fed data. In the case of non-string data
466 Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
467 TF_Tensor* TF_TensorFromTensor(const tensorflow::Tensor& src
471 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); local
481 Safe_TF_TensorPtr tf_tensor = make_safe(TF_TensorFromTensor(t, status)); local
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_test_util.h | 31 typedef std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)>
34 TF_Tensor* BoolTensor(int32_t v);
37 TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values);
40 TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
44 TF_Tensor* Int32Tensor(const std::vector<int32_t>& values);
46 TF_Tensor* Int32Tensor(int32_t v);
48 TF_Tensor* DoubleTensor(double v);
50 TF_Tensor* FloatTensor(float v);
57 TF_Operation* Const(TF_Tensor* t, TF_Graph* graph, TF_Status* s,
140 void SetInputs(std::vector<std::pair<TF_Operation*, TF_Tensor*>> inputs)
[all...] |
| kernels.h | 104 TF_Tensor** tensor, TF_Status* status);
111 const TF_Tensor* tensor,
|
| c_api.h | 206 // TF_Tensor holds a multi-dimensional array of elements of a single data type.
222 typedef struct TF_Tensor TF_Tensor;
234 TF_CAPI_EXPORT extern TF_Tensor* TF_NewTensor(
248 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType,
252 // Deletes `tensor` and returns a new TF_Tensor with the same content if
254 TF_CAPI_EXPORT extern TF_Tensor* TF_TensorMaybeMove(TF_Tensor* tensor);
257 TF_CAPI_EXPORT extern void TF_DeleteTensor(TF_Tensor*);
260 TF_CAPI_EXPORT extern TF_DataType TF_TensorType(const TF_Tensor*);
[all...] |
| c_api_internal.h | 57 struct TF_Tensor {
58 ~TF_Tensor();
206 Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
208 TF_Tensor* TF_TensorFromTensor(const Tensor& src, TF_Status* status);
|
| c_test_util.cc | 45 TF_Tensor* BoolTensor(bool v) {
53 TF_Tensor* Int8Tensor(const int64_t* dims, int num_dims, const char* values) {
58 TF_Tensor* t =
64 TF_Tensor* Int32Tensor(const int64_t* dims, int num_dims,
70 TF_Tensor* t =
76 TF_Tensor* Int32Tensor(const std::vector<int32_t>& values) {
81 TF_Tensor* Int32Tensor(int32_t v) {
89 TF_Tensor* DoubleTensor(double v) {
97 TF_Tensor* FloatTensor(float v) {
128 void ConstHelper(TF_Tensor* t, TF_Graph* graph, TF_Status* s, const char* name
[all...] |
| c_api_experimental.h | 125 TF_CAPI_EXPORT extern TF_Tensor* TF_DequeueNamedTensor(TF_Session* session,
142 TF_Tensor* tensor,
|
| c_test.c | 41 TF_Tensor* input;
|
| kernels.cc | 133 void TF_GetInput(TF_OpKernelContext* ctx, int i, TF_Tensor** tensor,
141 TF_Tensor* result = ::tensorflow::TF_TensorFromTensor(cc_tensor, status);
147 void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor,
|
| c_api.cc | 195 TF_Tensor::~TF_Tensor() { buffer->Unref(); }
197 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims,
204 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims,
219 // TF_Tensor than they do in tensorflow::Tensor. So a copy here is a waste
234 TF_Tensor* ret = new TF_Tensor{dtype, TensorShape(dimvec), buf};
243 TF_Tensor* TF_TensorMaybeMove(TF_Tensor* tensor) {
255 void TF_DeleteTensor(TF_Tensor* t) { delete t;
[all...] |
| c_api_test.cc | 53 TF_Tensor* TF_TensorFromTensor(const Tensor& src, TF_Status* status);
54 Status TF_TensorToTensor(const TF_Tensor* src, Tensor* dst);
133 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes,
152 TF_Tensor* t =
160 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, num_bytes);
177 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes,
180 TF_Tensor* o = TF_TensorMaybeMove(t);
238 TF_Tensor* dst = TF_TensorFromTensor(src, status);
[all...] |
| c_api_experimental_test.cc | 46 TF_Tensor* out = csession.output_tensor(0);
94 TF_Tensor* image = csession.output_tensor(0);
108 TF_Tensor* label = csession.output_tensor(1);
199 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status);
279 TF_Tensor* t = TFE_TensorHandleResolve(recv_retvals[0], status);
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_test_util.cc | 26 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(float));
38 TF_Tensor* t = TF_AllocateTensor(TF_INT32, nullptr, 0, sizeof(int));
50 TF_Tensor* t = TF_AllocateTensor(TF_BOOL, nullptr, 0, sizeof(bool));
63 TF_Tensor* t = TF_AllocateTensor(
77 TF_Tensor* t = TF_AllocateTensor(
91 TF_Tensor* t = TF_AllocateTensor(
105 TF_Tensor* t = TF_AllocateTensor(
147 TF_Tensor* t = TF_AllocateTensor(
|
| c_api.h | 154 // Like a TF_Tensor, a TFE_TensorHandle refers to a tensor with a value, shape,
155 // type etc. Unlike a TF_Tensor, a TFE_TensorHandle may refer to such tensors
159 TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandle(TF_Tensor* t,
198 // TF_Tensor).
199 TF_CAPI_EXPORT extern TF_Tensor* TFE_TensorHandleResolve(TFE_TensorHandle* h,
338 TF_Tensor* tensor,
|
| c_api_test.cc | 179 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status);
265 TF_Tensor* t = TFE_TensorHandleResolve(retval_task0, status);
304 TF_Tensor* t = TFE_TensorHandleResolve(handle, status.get());
447 TF_Tensor* t = TFE_TensorHandleResolve(h, status.get());
469 TF_Tensor* t = TFE_TensorHandleResolve(hcpu, status.get());
509 TF_Tensor* tcopy = TFE_TensorHandleResolve(hcopy, status.get());
583 TF_Tensor* t = TFE_TensorHandleResolve(hcpu, status.get());
630 TF_Tensor* tcopy = TFE_TensorHandleResolve(hcopy, status.get());
663 TF_Tensor* t = TFE_TensorHandleResolve(hcpu, status.get());
709 TF_Tensor* t = TFE_TensorHandleResolve(hcpu, status.get())
[all...] |
| c_api_experimental_test.cc | 89 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status);
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| session_jni.cc | 146 std::unique_ptr<TF_Tensor* []> input_values(new TF_Tensor*[ninputs]);
148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]);
|
| tensor_jni.cc | 29 TF_Tensor* requireHandle(JNIEnv* env, jlong handle) {
35 return reinterpret_cast<TF_Tensor*>(handle);
68 // Write a Java scalar object (java.lang.Integer etc.) to a TF_Tensor.
240 StringTensorWriter(TF_Tensor* t, int num_elements)
272 StringTensorReader(const TF_Tensor* t, int num_elements)
351 TF_Tensor* t = TF_AllocateTensor(static_cast<TF_DataType>(dtype), dims_copy,
371 TF_Tensor* t = TF_AllocateTensor(TF_STRING, nullptr, 0, 8 + dst_len);
454 TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims,
480 TF_DeleteTensor(reinterpret_cast<TF_Tensor*>(handle));
486 TF_Tensor* t = requireHandle(env, handle)
[all...] |
| operation_builder_jni.cc | 44 TF_Tensor* requireTensor(JNIEnv* env, jlong handle) {
50 return reinterpret_cast<TF_Tensor*>(handle);
211 TF_Tensor* t = requireTensor(env, tensor_handle);
227 std::unique_ptr<TF_Tensor* []> tensors(new TF_Tensor*[n]);
|
| /external/tensorflow/tensorflow/c/kernels/ |
| bitcast_op.cc | 93 TF_Tensor* tensor;
120 TF_Tensor* output = TF_AllocateTensor(k->output_data_type, dims, 0,
|
| /external/tensorflow/tensorflow/go/ |
| session.go | 356 feedTensors []*C.TF_Tensor
358 fetchTensors []*C.TF_Tensor
365 fetchTensors: make([]*C.TF_Tensor, len(fetches)),
396 func ptrTensor(l []*C.TF_Tensor) **C.TF_Tensor {
|
| attrs.go | 141 list := make([]*C.TF_Tensor, meta.list_size)
220 var v *C.TF_Tensor
|
| /external/tensorflow/tensorflow/python/client/ |
| tf_session_helper.h | 41 typedef gtl::InlinedVector<TF_Tensor*, 8> TF_TensorVector;
|
