Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc35 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
47 tstring* data = static_cast<tstring*>(TF_TensorData(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()
[all …]
Dmnist_gradients_test.cc112 memcpy(&result_data[0], TF_TensorData(dA_tensor), in TEST_P()
125 memcpy(&result_data[0], TF_TensorData(dB_tensor), in TEST_P()
189 memcpy(&result_data[0], TF_TensorData(scores_tensor), in TEST_P()
202 memcpy(&result_data[0], TF_TensorData(loss_vals_tensor), in TEST_P()
266 memcpy(&result_data[0], TF_TensorData(scores_tensor), in TEST_P()
279 memcpy(&result_data[0], TF_TensorData(loss_vals_tensor), in TEST_P()
335 memcpy(&result_data[0], TF_TensorData(scores_tensor), in TEST_P()
418 memcpy(&result_data[0], TF_TensorData(dW1_tensor), in TEST_P()
430 memcpy(&result_data[0], TF_TensorData(dW2_tensor), in TEST_P()
485 memcpy(&result_data[0], TF_TensorData(dA_tensor), in TEST_P()
Dgradient_checker.cc110 memcpy(theta_data.data(), TF_TensorData(theta_tensor), in CalcNumericalGrad()
139 memcpy(thetaPlus_data.data(), TF_TensorData(theta_tensor), in CalcNumericalGrad()
152 memcpy(&thetaMinus_data[0], TF_TensorData(theta_tensor), in CalcNumericalGrad()
190 memcpy(&grad_data[0], TF_TensorData(grad_tensor), in CalcNumericalGrad()
Dc_api_test.cc190 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
258 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
379 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
631 void* orig_ptr = TF_TensorData(orig); in ExecuteAdd()
656 EXPECT_EQ(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
663 EXPECT_EQ(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
665 EXPECT_NE(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
676 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteAdd()
744 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
882 memcpy(&output[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
[all …]
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.cc97 float* result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
165 memcpy(&result_data[0], TF_TensorData(result_tensor), in TEST_P()
248 memcpy(&result_data[0], TF_TensorData(result_tensor), in TEST_P()
334 memcpy(&result_data[0], TF_TensorData(result_tensor), in TEST_P()
427 float* f_value = static_cast<float*>(TF_TensorData(f_t)); in TEST_P()
528 memcpy(&result_data[0], TF_TensorData(f_t), TF_TensorByteSize(f_t)); in TEST_P()
687 results[idx] = *static_cast<float*>(TF_TensorData(f_t)); in TEST_P()
892 memcpy(&result_data[0], TF_TensorData(f_t), TF_TensorByteSize(f_t)); in TEST_P()
Dc_api_experimental_test.cc282 *reinterpret_cast<tensorflow::int32*>(TF_TensorData(t)) = 42; in TEST()
302 EXPECT_EQ(*reinterpret_cast<tensorflow::int32*>(TF_TensorData(r)), 42); in TEST()
353 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
378 float* m_float = static_cast<float*>(TF_TensorData(m_data)); in TEST()
401 static_cast<const float*>(TF_TensorData(resolved)); in TEST()
Dc_api_distributed_test.cc217 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TestFunctionWithPackedInput()
401 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
529 memcpy(&sum, TF_TensorData(t), TF_TensorByteSize(t)); in TestDistributedFunctionCancellation()
Dc_api_remote_test.cc90 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecute()
Dgradient_checker_test.cc55 memcpy(&dnumerical[0], TF_TensorData(numerical_tensor), in CompareNumericalAndManualGradients()
Dc_api_remote_test_util.cc199 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in TestRemoteExecuteSilentCopies()
/external/tensorflow/tensorflow/c/kernels/
Dsummary_op.cc90 static_cast<tensorflow::tstring*>(TF_TensorData(params.tags)); in ScalarSummaryOp_Compute()
91 auto values_array = static_cast<T*>(TF_TensorData(params.values)); in ScalarSummaryOp_Compute()
108 reinterpret_cast<tensorflow::tstring*>(TF_TensorData(summary_tensor)); in ScalarSummaryOp_Compute()
128 static_cast<tensorflow::tstring*>(TF_TensorData(tags))->c_str(); in SingleTag()
Dhistogram_summary_op.cc88 auto values_array = static_cast<T*>(TF_TensorData(safe_values_ptr.get())); in HistogramSummaryOp_Compute()
110 *(static_cast<tensorflow::tstring*>(TF_TensorData(safe_tags_ptr.get()))); in HistogramSummaryOp_Compute()
124 TF_TensorData(summary_tensor.get())); in HistogramSummaryOp_Compute()
Dmerge_summary_op.cc63 static_cast<tensorflow::tstring*>(TF_TensorData(safe_input_ptr.get())); in MergeSummaryOp_Compute()
97 TF_TensorData(summary_tensor.get())); in MergeSummaryOp_Compute()
/external/tensorflow/tensorflow/c/experimental/gradients/
Dgrad_test_helper.cc61 memcpy(&dnumerical[0], TF_TensorData(numerical_tensor), in CompareNumericalAndAutodiffGradients()
64 memcpy(&danalytical[0], TF_TensorData(analytical_tensor), in CompareNumericalAndAutodiffGradients()
94 memcpy(&danalytical[0], TF_TensorData(analytical_tensor), in CheckTensorValue()
Dcustom_gradient_test.cc123 auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc238 : index_(0), data_(static_cast<TF_TString*>(TF_TensorData(t))) {} in StringTensorWriter()
254 : index_(0), data_(static_cast<const TF_TString*>(TF_TensorData(t))) {} in StringTensorReader()
331 TF_TString* dst = static_cast<TF_TString*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_allocateScalarBytes()
437 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_buffer()
478 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_setValue()
500 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \
524 const TF_TString* data = static_cast<const TF_TString*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_scalarBytes()
537 const void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_readNDArray()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc139 float output_value = *static_cast<float*>(TF_TensorData(result)); in TEST_P()
279 float output_value = *static_cast<float*>(TF_TensorData(result)); in TEST_P()
336 static_cast<tensorflow::tstring*>(TF_TensorData(result)); in TEST_P()
405 static_cast<tensorflow::int64*>(TF_TensorData(result)); in TEST_P()
437 static_cast<tensorflow::int64*>(TF_TensorData(result)); in TEST_P()
470 static_cast<tensorflow::int64*>(TF_TensorData(result)); in TEST_P()
571 float output_value = *static_cast<float*>(TF_TensorData(result)); in TEST_P()
/external/llvm-project/llvm/lib/Analysis/
DTFUtils.cpp440 std::memset(TF_TensorData(Input[Index]), 0, TotalSize); in initInput()
444 return TF_TensorData(Impl->getInput()[Index]); in getUntypedInput()
461 return TF_TensorData(Impl->getOutput()[Index]); in getUntypedTensorValue()
466 return TF_TensorData(Impl->getOutput()[Index]); in getUntypedTensorValue()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc275 const void* tensor_data = TF_TensorData(src); in CopyTF_TensorStringsToPyArray()
409 dims.size(), dims.data(), TF_TensorData(moved), in TF_TensorToMaybeAliasedPyArray()
434 dims.size(), dims.data(), TF_TensorData(moved), in TF_TensorToPyArray()
464 FastMemcpy(PyArray_DATA(py_array), TF_TensorData(tensor.get()), in TF_TensorToPyArray()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc144 EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t)); in TEST()
1007 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
1024 output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
1068 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in RunMinTest()
1142 EXPECT_EQ(3, *(static_cast<int32*>(TF_TensorData(fetchValues1[0])))); in TEST()
1154 EXPECT_EQ(7, *(static_cast<int32*>(TF_TensorData(fetchValues2[0])))); in TEST()
1452 float* values = static_cast<float*>(TF_TensorData(out)); in TEST()
1535 *(static_cast<int64_t*>(TF_TensorData(a))) = 4; in TEST()
1536 EXPECT_EQ(4, *(static_cast<int64_t*>(TF_TensorData(b)))); in TEST()
1537 *(static_cast<int64_t*>(TF_TensorData(b))) = 6; in TEST()
[all …]
Dtf_tensor.h127 TF_CAPI_EXPORT extern void* TF_TensorData(const TF_Tensor*);
Dtf_tensor.cc159 void* TF_TensorData(const TF_Tensor* t) { return t->tensor->Data(); } in TF_TensorData() function
283 std::memcpy(TF_TensorData(t), str.c_str(), str.size()); in TF_TensorFromTensor()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensor.h127 inline void* Tensor::data() const { return TF_TensorData(tensor_.get()); } in data()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.h149 *static_cast<value_type*>(TF_TensorData(actual_value.get()))); in ExpectScalarEq()

12