Home
last modified time | relevance | path

Searched refs:TF_TensorByteSize (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc38 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
63 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
75 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
89 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle()
103 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle()
117 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandleWithInput()
130 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestTensorHandleWithDimsFloat()
143 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestTensorHandleWithDimsInt()
162 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle100x100()
176 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle3X2()
[all …]
Dgradient_checker.cc112 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
141 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
154 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
192 TF_TensorByteSize(grad_tensor)); in CalcNumericalGrad()
Dc_api_distributed_test.cc216 EXPECT_EQ(sizeof(sum), TF_TensorByteSize(t)); in TestFunctionWithPackedInput()
217 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TestFunctionWithPackedInput()
400 ASSERT_EQ(sizeof(sum), TF_TensorByteSize(t)); in TEST()
401 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
528 ASSERT_EQ(sizeof(sum), TF_TensorByteSize(t)); in TestDistributedFunctionCancellation()
529 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TestDistributedFunctionCancellation()
Dc_api_test.cc186 ASSERT_EQ(16, TF_TensorByteSize(t)); in TEST()
188 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
254 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)) << tag; in TensorHandleCopyBetweenDevices()
256 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
375 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)); in TensorHandleCopyBetweenTwoGPUDevices()
377 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
702 EXPECT_EQ(sizeof(result), TF_TensorByteSize(t)); in ExecuteAdd()
703 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteAdd()
771 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
772 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
[all …]
Dc_api_remote_test.cc89 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecute()
90 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecute()
Dc_api_remote_test_util.cc198 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
199 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
Dc_api_test_util.h65 memcpy(TF_TensorData(t), data, TF_TensorByteSize(t)); in TestTensorHandleWithDims()
Dc_api_unified_experimental_test.cc166 TF_TensorByteSize(result_tensor)); in TEST_P()
249 TF_TensorByteSize(result_tensor)); in TEST_P()
335 TF_TensorByteSize(result_tensor)); in TEST_P()
528 memcpy(&result_data[0], TF_TensorData(f_t), TF_TensorByteSize(f_t)); in TEST_P()
892 memcpy(&result_data[0], TF_TensorData(f_t), TF_TensorByteSize(f_t)); in TEST_P()
Dgradient_checker_test.cc56 TF_TensorByteSize(numerical_tensor)); in CompareNumericalAndManualGradients()
Dc_api_cluster_test.cc47 EXPECT_EQ(sizeof(float) * expected_values.size(), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
48 memcpy(actual_values.get(), TF_TensorData(t), TF_TensorByteSize(t)); in CheckTFE_TensorHandleHasFloats()
Dc_api_experimental_test.cc355 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
356 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dgrad_test_helper.cc62 TF_TensorByteSize(numerical_tensor)); in CompareNumericalAndAutodiffGradients()
65 TF_TensorByteSize(analytical_tensor)); in CompareNumericalAndAutodiffGradients()
95 TF_TensorByteSize(analytical_tensor)); in CheckTensorValue()
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_multi_client_test.cc108 EXPECT_EQ(sizeof(result), TF_TensorByteSize(t)); in TEST()
109 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
Dc_api_multi_client_function_test.cc337 EXPECT_EQ(sizeof(result), TF_TensorByteSize(t)); in TEST_P()
338 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST_P()
/external/tensorflow/tensorflow/c/
Dtf_tensor.h128 TF_CAPI_EXPORT extern size_t TF_TensorByteSize(const TF_Tensor*);
Dc_api_test.cc147 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
172 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
1009 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
1026 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
1070 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in RunMinTest()
1523 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1524 EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1534 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1535 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1921 ASSERT_EQ(4, TF_TensorByteSize(t)); in ScalarFloatFromTensor()
[all …]
Dtf_tensor.cc162 size_t TF_TensorByteSize(const TF_Tensor* t) { return t->tensor->ByteSize(); } in TF_TensorByteSize() function
Dwhile_loop_test.cc100 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in ExpectOutputValue()
Dc_api_function_test.cc130 ASSERT_EQ(sizeof(int32_t), TF_TensorByteSize(out)); in Run()
138 size_t size = TF_TensorByteSize(t); in CompareInt32Tensor()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensor.h144 return TF_TensorByteSize(tensor_.get()); in num_bytes()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc324 dims->push_back(TF_TensorByteSize(tensor)); in GetPyArrayDimensionsForTensor()
466 TF_TensorByteSize(tensor.get())) { in TF_TensorToPyArray()
469 TF_TensorByteSize(tensor.get()), " bytes"); in TF_TensorToPyArray()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc442 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_buffer()
483 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_setValue()
542 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/go/
Dtensor.go382 length := int(C.TF_TensorByteSize(c))
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc182 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in SetTensorProto()