| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 27 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()
|
| D | c_api_test.cc | 182 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecute()
183 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecute()
269 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
270 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
307 EXPECT_EQ(sizeof(float) * expected_values.size(), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
308 memcpy(actual_values.get(), TF_TensorData(t), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
448 ASSERT_EQ(16, TF_TensorByteSize(t)); in TEST()
450 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
515 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)) << tag; in TensorHandleCopyBetweenDevices()
517 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
[all …]
|
| D | c_api_experimental_test.cc | 94 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in ExecuteWithProfiling()
95 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteWithProfiling()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 50 ASSERT_EQ(sizeof(float), TF_TensorByteSize(out)); in TestFakeIteratorStack()
104 TF_TensorByteSize(image)); in TEST()
113 ASSERT_EQ(sizeof(int32) * batch_size, TF_TensorByteSize(label)); in TEST()
202 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TEST()
203 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
283 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TEST()
284 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
|
| D | c_api_test.cc | 140 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
165 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
969 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
986 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
1030 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in RunMinTest()
1482 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1483 EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1493 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1494 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1852 ASSERT_EQ(4, TF_TensorByteSize(t)); in ScalarFloatFromTensor()
[all …]
|
| D | while_loop_test.cc | 101 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in ExpectOutputValue()
|
| D | c_api_function_test.cc | 130 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in Run()
138 size_t size = TF_TensorByteSize(t); in CompareInt32Tensor()
1719 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST_F()
|
| D | c_api.h | 270 TF_CAPI_EXPORT extern size_t TF_TensorByteSize(const TF_Tensor*);
|
| D | c_api.cc | 262 size_t TF_TensorByteSize(const TF_Tensor* t) { return t->buffer->size(); } in TF_TensorByteSize() function
501 TF_TensorByteSize(src)))) { in TF_TensorToTensor()
515 const size_t src_size = TF_TensorByteSize(src); in TF_TensorToTensor()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 244 plimit_(poffsets_ + TF_TensorByteSize(t)) {} in StringTensorWriter()
276 limit_(offsets_ + TF_TensorByteSize(t)) {} in StringTensorReader()
489 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_buffer()
530 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_setValue()
577 size_t src_len = TF_TensorByteSize(t) - 8; in Java_org_tensorflow_Tensor_scalarBytes()
601 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_readNDArray()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 223 const size_t tensor_size = TF_TensorByteSize(src); in CopyTF_TensorStringsToPyArray()
279 dims.push_back(TF_TensorByteSize(tensor)); in GetPyArrayDimensionsForTensor()
401 TF_TensorByteSize(tensor.get())) { in TF_TensorToPyArray()
404 TF_TensorByteSize(tensor.get()), " bytes"); in TF_TensorToPyArray()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 89 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in SetTensorProto()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 209 length := int(C.TF_TensorByteSize(c))
|