Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc35 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
60 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
72 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
86 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle()
100 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle()
114 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandleWithInput()
127 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestTensorHandleWithDimsFloat()
140 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestTensorHandleWithDimsInt()
159 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestMatrixTensorHandle100x100()
173 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in DoubleTestMatrixTensorHandle3X2()
[all …]
Dmnist_gradients_test.cc113 TF_TensorByteSize(dA_tensor)); in TEST_P()
126 TF_TensorByteSize(dB_tensor)); in TEST_P()
190 TF_TensorByteSize(scores_tensor)); in TEST_P()
203 TF_TensorByteSize(loss_vals_tensor)); in TEST_P()
267 TF_TensorByteSize(scores_tensor)); in TEST_P()
280 TF_TensorByteSize(loss_vals_tensor)); in TEST_P()
336 TF_TensorByteSize(scores_tensor)); in TEST_P()
419 TF_TensorByteSize(dW1_tensor)); in TEST_P()
431 TF_TensorByteSize(dW2_tensor)); in TEST_P()
486 TF_TensorByteSize(dA_tensor)); in TEST_P()
Dgradient_checker.cc111 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
140 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
153 TF_TensorByteSize(theta_tensor)); in CalcNumericalGrad()
191 TF_TensorByteSize(grad_tensor)); in CalcNumericalGrad()
Dc_api_test.cc188 ASSERT_EQ(16, TF_TensorByteSize(t)); in TEST()
190 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
256 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)) << tag; in TensorHandleCopyBetweenDevices()
258 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
377 EXPECT_EQ(TF_TensorByteSize(t), TF_TensorByteSize(tcopy)); in TensorHandleCopyBetweenTwoGPUDevices()
379 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
675 EXPECT_EQ(sizeof(result), TF_TensorByteSize(t)); in ExecuteAdd()
676 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteAdd()
743 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
744 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
[all …]
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_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()
Dgradients_util.cc47 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in ScalarTensorHandleHelper()
61 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TensorHandleWithDimsFloatHelper()
75 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TensorHandleWithDimsIntHelper()
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.cc352 EXPECT_EQ(sizeof(product), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
353 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/c/
Dtf_tensor.h124 TF_CAPI_EXPORT extern size_t TF_TensorByteSize(const TF_Tensor*);
Dc_api_test.cc143 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
168 EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); in TEST()
1006 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
1023 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in TEST()
1067 ASSERT_EQ(sizeof(int32), TF_TensorByteSize(out)); in RunMinTest()
1520 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1521 EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1531 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a)); in TEST()
1532 EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b)); in TEST()
1890 ASSERT_EQ(4, TF_TensorByteSize(t)); in ScalarFloatFromTensor()
[all …]
Dtf_tensor.cc157 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.cc316 dims->push_back(TF_TensorByteSize(tensor)); in GetPyArrayDimensionsForTensor()
459 TF_TensorByteSize(tensor.get())) { in TF_TensorToPyArray()
462 TF_TensorByteSize(tensor.get()), " bytes"); in TF_TensorToPyArray()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc438 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_buffer()
479 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_setValue()
538 const size_t sz = TF_TensorByteSize(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/go/
Dtensor.go379 length := int(C.TF_TensorByteSize(c))
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc176 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in SetTensorProto()