Home
last modified time | relevance | path

Searched refs:AbstractTensorHandlePtr (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dgradient_checker.cc58 AbstractTensorHandlePtr model_out(model_outputs[0]); in RunAndMaybeSum()
74 AbstractTensorHandlePtr sum_dims; in RunAndMaybeSum()
131 AbstractTensorHandlePtr two_eps; in CalcNumericalGrad()
143 AbstractTensorHandlePtr thetaPlus; in CalcNumericalGrad()
156 AbstractTensorHandlePtr thetaMinus; in CalcNumericalGrad()
169 AbstractTensorHandlePtr fPlus(f_outputs[0]); in CalcNumericalGrad()
175 AbstractTensorHandlePtr fMinus(f_outputs[0]); in CalcNumericalGrad()
180 AbstractTensorHandlePtr fDiff(f_outputs[0]); in CalcNumericalGrad()
186 AbstractTensorHandlePtr diff_quotient(f_outputs[0]); in CalcNumericalGrad()
Dgradient_checker_test.cc39 AbstractTensorHandlePtr numerical_grad; in CompareNumericalAndManualGradients()
120 AbstractTensorHandlePtr A; in TEST_P()
130 AbstractTensorHandlePtr B; in TEST_P()
146 AbstractTensorHandlePtr x; in TEST_P()
155 AbstractTensorHandlePtr y; in TEST_P()
Dunified_api_test.cc68 AbstractTensorHandlePtr x; in TEST_P()
118 AbstractTensorHandlePtr x; in TEST_P()
145 AbstractTensorHandlePtr x; in TEST_P()
169 AbstractTensorHandlePtr x; in TEST_P()
Dgradients_test.cc77 AbstractTensorHandlePtr t; in TEST_P()
143 AbstractTensorHandlePtr x; in TEST_P()
Dabstract_tensor_handle.h62 using AbstractTensorHandlePtr = variable
/external/tensorflow/tensorflow/c/experimental/gradients/
Dmath_grad_test.cc115 AbstractTensorHandlePtr x; in TEST_P()
124 AbstractTensorHandlePtr y; in TEST_P()
144 AbstractTensorHandlePtr x; in TEST_P()
169 AbstractTensorHandlePtr A; in TEST_P()
180 AbstractTensorHandlePtr B; in TEST_P()
212 AbstractTensorHandlePtr A; in TEST_P()
223 AbstractTensorHandlePtr B; in TEST_P()
275 AbstractTensorHandlePtr x; in TEST_P()
293 AbstractTensorHandlePtr x; in TEST_P()
311 AbstractTensorHandlePtr x; in TEST_P()
[all …]
Dmath_grad.cc80 AbstractTensorHandlePtr conj_output_releaser(conj_output); in Compute()
90 AbstractTensorHandlePtr exp_;
109 AbstractTensorHandlePtr sqrt_;
149 AbstractTensorHandlePtr A(conj_output); in Compute()
155 AbstractTensorHandlePtr B(conj_output); in Compute()
342 AbstractTensorHandlePtr Conj_X(temp_output); in Compute()
348 AbstractTensorHandlePtr Ones_X(temp_output); in Compute()
355 AbstractTensorHandlePtr Conj_XP1(temp_output); in Compute()
419 AbstractTensorHandlePtr MinusU(temp_output); in Compute()
424 AbstractTensorHandlePtr UZ(temp_output); in Compute()
Dnn_grad_test.cc103 AbstractTensorHandlePtr X; in TEST_P()
118 AbstractTensorHandlePtr Y; in TEST_P()
146 AbstractTensorHandlePtr X; in TEST_P()
157 AbstractTensorHandlePtr Y; in TEST_P()
184 AbstractTensorHandlePtr A; in TEST_P()
195 AbstractTensorHandlePtr Bias; in TEST_P()
Darray_grad_test.cc84 AbstractTensorHandlePtr x1; in TEST_P()
93 AbstractTensorHandlePtr x2; in TEST_P()
Dcustom_gradient_test.cc99 AbstractTensorHandlePtr x; in TEST_P()
Dgrad_test_helper.cc38 AbstractTensorHandlePtr numerical_grad; in CompareNumericalAndAutodiffGradients()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc61 AbstractTensorHandlePtr owned_var_handle(var_handle); in CreateUninitializedResourceVariable()
97 AbstractTensorHandlePtr owned_value(value); in ReadVariable()
Drestore_ops.cc100 AbstractTensorHandlePtr owned_restored_handle(restored_handle); in SingleRestore()
/external/tensorflow/tensorflow/cc/experimental/libtf/tests/
Dvariable_test.cc87 tensorflow::AbstractTensorHandlePtr var; in TEST_P()
102 tensorflow::AbstractTensorHandlePtr value; in TEST_P()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Drestored_resource.cc40 AbstractTensorHandlePtr owned_dummy_output(dummy_output); in ExecuteNoArgDummyReturnFunction()
Dpartially_revived_objects.cc416 AbstractTensorHandlePtr owned_resource_handle(resource_handle); in CreateAllResourceHandles()
/external/tensorflow/tensorflow/python/framework/experimental/
Dunified_api.cc52 using tensorflow::AbstractTensorHandlePtr;
228 py::class_<AbstractTensorHandle, AbstractTensorHandlePtr>( in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache_test.cc190 tensorflow::AbstractTensorHandlePtr x; in TEST_P()
198 tensorflow::AbstractTensorHandlePtr y; in TEST_P()