Home
last modified time | relevance | path

Searched refs:TF_TensorData (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc27 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
39 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
51 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
65 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle()
79 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle()
93 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle3X2()
107 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle3X2()
149 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestAxisTensorHandle()
Dc_api_test.cc183 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecute()
270 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
308 memcpy(actual_values.get(), TF_TensorData(t), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
450 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
517 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
635 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
898 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
1031 memcpy(&output[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
1071 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_XLA_CPU()
1112 memcpy(&output[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_Min_XLA_CPU()
[all …]
Dc_api_experimental_test.cc95 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteWithProfiling()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc242 poffsets_(static_cast<char*>(TF_TensorData(t))), in StringTensorWriter()
274 offsets_(static_cast<const char*>(TF_TensorData(t))), in StringTensorReader()
372 char* dst = static_cast<char*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_allocateScalarBytes()
488 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_buffer()
529 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_setValue()
551 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \
575 const char* data = static_cast<const char*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_scalarBytes()
600 const void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc141 EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t)); in TEST()
970 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
987 output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
1031 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in RunMinTest()
1105 EXPECT_EQ(3, *(static_cast<int32*>(TF_TensorData(fetchValues1[0])))); in TEST()
1117 EXPECT_EQ(7, *(static_cast<int32*>(TF_TensorData(fetchValues2[0])))); in TEST()
1414 float* values = static_cast<float*>(TF_TensorData(out)); in TEST()
1497 *(static_cast<int64_t*>(TF_TensorData(a))) = 4; in TEST()
1498 EXPECT_EQ(4, *(static_cast<int64_t*>(TF_TensorData(b)))); in TEST()
1499 *(static_cast<int64_t*>(TF_TensorData(b))) = 6; in TEST()
[all …]
Dc_api_experimental_test.cc51 float* output_contents = static_cast<float*>(TF_TensorData(out)); in TestFakeIteratorStack()
203 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
284 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
Dkernels_test.cc165 EXPECT_EQ(123, *static_cast<tensorflow::uint8*>(TF_TensorData(input))); in TEST()
Dc_test_util.cc60 memcpy(TF_TensorData(t), values, sizeof(char) * num_values); in Int8Tensor()
72 memcpy(TF_TensorData(t), values, sizeof(int32_t) * num_values); in Int32Tensor()
Dwhile_loop_test.cc102 int32_t* data = static_cast<int32_t*>(TF_TensorData(out)); in ExpectOutputValue()
Dc_api_function_test.cc131 int32_t* output_contents = static_cast<int32_t*>(TF_TensorData(out)); in Run()
137 int32_t* data = static_cast<int32_t*>(TF_TensorData(t)); in CompareInt32Tensor()
1720 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST_F()
Dc_api.cc263 void* TF_TensorData(const TF_Tensor* t) { return t->buffer->data(); } in TF_TensorData() function
500 string(static_cast<const char*>(TF_TensorData(src)), in TF_TensorToTensor()
514 const char* input = reinterpret_cast<const char*>(TF_TensorData(src)); in TF_TensorToTensor()
586 std::memcpy(TF_TensorData(t), str.c_str(), str.size()); in TF_TensorFromTensor()
Dc_api.h273 TF_CAPI_EXPORT extern void* TF_TensorData(const TF_Tensor*);
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc222 const void* tensor_data = TF_TensorData(src); in CopyTF_TensorStringsToPyArray()
376 if (ArrayFromMemory(dims.size(), dims.data(), TF_TensorData(moved), in TF_TensorToPyArray()
406 FastMemcpy(PyArray_DATA(py_array), TF_TensorData(tensor.get()), in TF_TensorToPyArray()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc89 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in SetTensorProto()
/external/tensorflow/tensorflow/go/
Dtensor.go205 cbytes := C.TF_TensorData(c)
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc1001 int32_t* data = reinterpret_cast<int32_t*>(TF_TensorData(tensor.get())); in TFE_Py_TensorShapeSlice()