Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc38 memcpy(TF_TensorData(t), &data[0], TF_TensorByteSize(t)); in TestScalarTensorHandle()
50 tstring* data = static_cast<tstring*>(TF_TensorData(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()
[all …]
Dgradient_checker.cc111 memcpy(theta_data.data(), TF_TensorData(theta_tensor), in CalcNumericalGrad()
140 memcpy(thetaPlus_data.data(), TF_TensorData(theta_tensor), in CalcNumericalGrad()
153 memcpy(&thetaMinus_data[0], TF_TensorData(theta_tensor), in CalcNumericalGrad()
191 memcpy(&grad_data[0], TF_TensorData(grad_tensor), in CalcNumericalGrad()
Dc_api_test.cc188 memcpy(&data[0], TF_TensorData(t), TF_TensorByteSize(t)); in TEST()
256 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))) in TensorHandleCopyBetweenDevices()
377 0, memcmp(TF_TensorData(t), TF_TensorData(tcopy), TF_TensorByteSize(t))); in TensorHandleCopyBetweenTwoGPUDevices()
634 void* orig_ptr = TF_TensorData(orig); in ExecuteAdd()
676 EXPECT_EQ(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
681 EXPECT_EQ(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
687 EXPECT_EQ(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
692 EXPECT_NE(orig_ptr, TF_TensorData(t)); in ExecuteAdd()
703 memcpy(&result[0], TF_TensorData(t), TF_TensorByteSize(t)); in ExecuteAdd()
772 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Execute_MatMul_CPU()
[all …]
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.cc284 *reinterpret_cast<tensorflow::int32*>(TF_TensorData(t)) = 42; in TEST()
304 EXPECT_EQ(*reinterpret_cast<tensorflow::int32*>(TF_TensorData(r)), 42); in TEST()
356 memcpy(&product[0], TF_TensorData(t), TF_TensorByteSize(t)); in Executor_MatMul_CPU()
381 float* m_float = static_cast<float*>(TF_TensorData(m_data)); in TEST()
404 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()
Dc_api_test_util.h65 memcpy(TF_TensorData(t), data, TF_TensorByteSize(t)); in TestTensorHandleWithDims()
Dgradient_checker_test.cc55 memcpy(&dnumerical[0], TF_TensorData(numerical_tensor), in CompareNumericalAndManualGradients()
/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/dtensor/cc/
Dsmall_constant_optimization.cc109 static_cast<int*>(TF_TensorData(value_tensor.get())), in ExtractSmallTensorValue()
115 static_cast<const int64_t*>(TF_TensorData(value_tensor.get())), in ExtractSmallTensorValue()
121 static_cast<const TF_TString*>(TF_TensorData(value_tensor.get())), in ExtractSmallTensorValue()
127 static_cast<const float*>(TF_TensorData(value_tensor.get())), in ExtractSmallTensorValue()
/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()
333 TF_TString* dst = static_cast<TF_TString*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_allocateScalarBytes()
441 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_buffer()
482 void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_setValue()
504 memcpy(&ret, TF_TensorData(t), elemByteSize(dtype)); \
528 const TF_TString* data = static_cast<const TF_TString*>(TF_TensorData(t)); in Java_org_tensorflow_Tensor_scalarBytes()
541 const void* data = TF_TensorData(t); in Java_org_tensorflow_Tensor_readNDArray()
/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.cc121 auto result_value = static_cast<float*>(TF_TensorData(result_tensor)); in TEST_P()
/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()
404 int64_t* output_value = static_cast<int64_t*>(TF_TensorData(result)); in TEST_P()
435 int64_t* output_value = static_cast<int64_t*>(TF_TensorData(result)); in TEST_P()
467 int64_t* output_value = static_cast<int64_t*>(TF_TensorData(result)); in TEST_P()
568 float output_value = *static_cast<float*>(TF_TensorData(result)); in TEST_P()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc148 EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t)); in TEST()
1010 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
1027 output_contents = static_cast<int32*>(TF_TensorData(out)); in TEST()
1071 int32* output_contents = static_cast<int32*>(TF_TensorData(out)); in RunMinTest()
1145 EXPECT_EQ(3, *(static_cast<int32*>(TF_TensorData(fetchValues1[0])))); in TEST()
1157 EXPECT_EQ(7, *(static_cast<int32*>(TF_TensorData(fetchValues2[0])))); in TEST()
1455 float* values = static_cast<float*>(TF_TensorData(out)); in TEST()
1538 *(static_cast<int64_t*>(TF_TensorData(a))) = 4; in TEST()
1539 EXPECT_EQ(4, *(static_cast<int64_t*>(TF_TensorData(b)))); in TEST()
1540 *(static_cast<int64_t*>(TF_TensorData(b))) = 6; in TEST()
[all …]
Dtf_tensor.h131 TF_CAPI_EXPORT extern void* TF_TensorData(const TF_Tensor*);
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor.cc283 const void* tensor_data = TF_TensorData(src); in CopyTF_TensorStringsToPyArray()
416 dims.size(), dims.data(), TF_TensorData(moved), in TF_TensorToMaybeAliasedPyArray()
441 dims.size(), dims.data(), TF_TensorData(moved), in TF_TensorToPyArray()
471 FastMemcpy(PyArray_DATA(py_array), TF_TensorData(tensor.get()), in TF_TensorToPyArray()
/external/tensorflow/tensorflow/cc/experimental/base/public/
Dtensor.h127 inline void* Tensor::data() const { return TF_TensorData(tensor_.get()); } in data()
/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
Dvariable_test.cc80 auto value = static_cast<T*>(TF_TensorData(result_t)); in ExpectEquals()
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_coordination_test.cc239 const tstring& next_val = *static_cast<tstring*>(TF_TensorData(t)); in TEST()
521 const tstring& get_val = *static_cast<tstring*>(TF_TensorData(t)); in TEST_P()
568 const tstring& get_fn_val = *static_cast<tstring*>(TF_TensorData(t)); in TEST_P()
Dc_api_session_coordination_test.cc157 *static_cast<tstring*>(TF_TensorData(output_values[0])); in TEST_P()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.h149 *static_cast<value_type*>(TF_TensorData(actual_value.get()))); in ExpectScalarEq()

12