Home
last modified time | relevance | path

Searched refs:TF_TensorByteSize (Results 1 – 14 of 14) 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()
97 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle100x100()
111 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle3X2()
125 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle3X2()
180 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestAxisTensorHandle()
Dc_api_test.cc159 ASSERT_EQ(16, TF_TensorByteSize(t)); in TEST()
161 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
226 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)) << tag; in TensorHandleCopyBetweenDevices()
228 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
347 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)); in TensorHandleCopyBetweenTwoGPUDevices()
349 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
617 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
618 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
755 EXPECT_EQ(sizeof(output), TF_TensorByteSize(t)); in TEST()
756 memcpy(&output[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
[all …]
Dc_api_remote_test.cc99 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecute()
100 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecute()
188 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
189 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
299 EXPECT_EQ(sizeof(float) * expected_values.size(), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
300 memcpy(actual_values.get(), TF_TensorData(t), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
Dc_api_experimental_test.cc101 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in ExecuteWithProfiling()
102 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteWithProfiling()
486 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
487 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc110 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TEST()
111 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
191 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TEST()
192 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
Dtf_tensor.h113 TF_CAPI_EXPORT extern size_t TF_TensorByteSize(const TF_Tensor*);
Dc_api_test.cc141 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
166 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()
1483 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1484 EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1494 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1495 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1853 ASSERT_EQ(4, TF_TensorByteSize(t)); in ScalarFloatFromTensor()
[all …]
Dwhile_loop_test.cc101 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in ExpectOutputValue()
Dtf_tensor.cc134 size_t TF_TensorByteSize(const TF_Tensor* t) { return t->tensor->ByteSize(); } in TF_TensorByteSize() function
Dc_api_function_test.cc130 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in Run()
138 size_t size = TF_TensorByteSize(t); in CompareInt32Tensor()
1756 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST_F()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc244 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/
Dndarray_tensor.cc276 const size_t tensor_size = TF_TensorByteSize(src); in CopyTF_TensorStringsToPyArray()
335 dims->push_back(TF_TensorByteSize(tensor)); in GetPyArrayDimensionsForTensor()
478 TF_TensorByteSize(tensor.get())) { in TF_TensorToPyArray()
481 TF_TensorByteSize(tensor.get()), " bytes"); in TF_TensorToPyArray()
/external/tensorflow/tensorflow/go/
Dtensor.go209 length := int(C.TF_TensorByteSize(c))
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc164 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in SetTensorProto()