Searched refs:TF_AllocateTensor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 26 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(float)); in TestScalarTensorHandle()
38 TF_Tensor* t = TF_AllocateTensor(TF_INT32, nullptr, 0, sizeof(int)); in TestScalarTensorHandle()
50 TF_Tensor* t = TF_AllocateTensor(TF_BOOL, nullptr, 0, sizeof(bool)); in TestScalarTensorHandle()
63 TF_Tensor* t = TF_AllocateTensor( in DoubleTestMatrixTensorHandle()
77 TF_Tensor* t = TF_AllocateTensor( in TestMatrixTensorHandle()
91 TF_Tensor* t = TF_AllocateTensor( in DoubleTestMatrixTensorHandle3X2()
105 TF_Tensor* t = TF_AllocateTensor( in TestMatrixTensorHandle3X2()
147 TF_Tensor* t = TF_AllocateTensor( in TestAxisTensorHandle()
|
| D | c_api_test.cc | 1202 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32)); in TEST()
1269 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32)); in TEST()
1446 TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(value)), TF_DeleteTensor); in CreateVariable()
1562 TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float)); in TEST()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 120 TF_Tensor* output = TF_AllocateTensor(k->output_data_type, dims, 0, in BitcastOp_Compute()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 351 TF_Tensor* t = TF_AllocateTensor(static_cast<TF_DataType>(dtype), dims_copy, in Java_org_tensorflow_Tensor_allocate()
371 TF_Tensor* t = TF_AllocateTensor(TF_STRING, nullptr, 0, 8 + dst_len); in Java_org_tensorflow_Tensor_allocateScalarBytes()
454 TF_Tensor* t = TF_AllocateTensor(TF_STRING, dims, num_dims, in Java_org_tensorflow_Tensor_allocateNonScalarBytes()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 90 …c: C.TF_AllocateTensor(C.TF_DataType(dataType), shapePtr, C.int(len(shape)), C.size_t(nbytes)),
129 …c: C.TF_AllocateTensor(C.TF_DataType(dataType), shapePtr, C.int(len(shape)), C.size_t(nbytes)),
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 59 TF_AllocateTensor(TF_INT8, dims, num_dims, sizeof(char) * num_values); in Int8Tensor()
71 TF_AllocateTensor(TF_INT32, dims, num_dims, sizeof(int32_t) * num_values); in Int32Tensor()
|
| D | c_api_test.cc | 160 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, num_bytes); in TEST()
1474 TF_AllocateTensor(TF_UINT64, dims, 2, 6 * TF_DataTypeSize(TF_UINT64)); in TEST()
1476 TF_AllocateTensor(TF_UINT64, nullptr, 0, TF_DataTypeSize(TF_UINT64)); in TEST()
1712 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, sizeof(float) * 4); in FloatTensor2x2()
|
| D | c_api.h | 248 TF_CAPI_EXPORT extern TF_Tensor* TF_AllocateTensor(TF_DataType,
|
| D | c_api.cc | 197 TF_Tensor* TF_AllocateTensor(TF_DataType dtype, const int64_t* dims, in TF_AllocateTensor() function
585 TF_Tensor* t = TF_AllocateTensor(TF_RESOURCE, {}, 0, str.size()); in TF_TensorFromTensor()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 87 TF_Tensor* t = TF_AllocateTensor( in SetTensorProto()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 999 auto tensor = tensorflow::make_safe(TF_AllocateTensor( in TFE_Py_TensorShapeSlice()
|