Searched refs:AbstractTensorHandlePtr (Results 1 – 18 of 18) sorted by relevance
57 AbstractTensorHandlePtr model_out(model_outputs[0]); in RunAndMaybeSum()73 AbstractTensorHandlePtr sum_dims; in RunAndMaybeSum()130 AbstractTensorHandlePtr two_eps; in CalcNumericalGrad()142 AbstractTensorHandlePtr thetaPlus; in CalcNumericalGrad()155 AbstractTensorHandlePtr thetaMinus; in CalcNumericalGrad()168 AbstractTensorHandlePtr fPlus(f_outputs[0]); in CalcNumericalGrad()174 AbstractTensorHandlePtr fMinus(f_outputs[0]); in CalcNumericalGrad()179 AbstractTensorHandlePtr fDiff(f_outputs[0]); in CalcNumericalGrad()185 AbstractTensorHandlePtr diff_quotient(f_outputs[0]); in CalcNumericalGrad()
84 AbstractTensorHandlePtr A = in TEST_P()86 AbstractTensorHandlePtr B = in TEST_P()153 AbstractTensorHandlePtr X = in TEST_P()158 AbstractTensorHandlePtr W1 = in TEST_P()163 AbstractTensorHandlePtr W2 = in TEST_P()170 AbstractTensorHandlePtr y = in TEST_P()229 AbstractTensorHandlePtr X = in TEST_P()235 AbstractTensorHandlePtr W1 = in TEST_P()240 AbstractTensorHandlePtr W2 = in TEST_P()247 AbstractTensorHandlePtr y = in TEST_P()[all …]
39 AbstractTensorHandlePtr numerical_grad; in CompareNumericalAndManualGradients()117 AbstractTensorHandlePtr A; in TEST_P()127 AbstractTensorHandlePtr B; in TEST_P()143 AbstractTensorHandlePtr x; in TEST_P()151 AbstractTensorHandlePtr y; in TEST_P()
55 AbstractTensorHandlePtr GetTensorHandleUtilFloat(AbstractContext* ctx,60 AbstractTensorHandlePtr GetTensorHandleUtilInt(AbstractContext* ctx, int vals[],64 AbstractTensorHandlePtr GetScalarTensorHandleUtil(AbstractContext* ctx,
68 AbstractTensorHandlePtr x; in TEST_P()118 AbstractTensorHandlePtr x; in TEST_P()145 AbstractTensorHandlePtr x; in TEST_P()169 AbstractTensorHandlePtr x; in TEST_P()
140 AbstractTensorHandlePtr GetTensorHandleUtilFloat(AbstractContext* ctx, in GetTensorHandleUtilFloat()143 AbstractTensorHandlePtr A; in GetTensorHandleUtilFloat()152 AbstractTensorHandlePtr GetTensorHandleUtilInt(AbstractContext* ctx, int vals[], in GetTensorHandleUtilInt()154 AbstractTensorHandlePtr A; in GetTensorHandleUtilInt()163 AbstractTensorHandlePtr GetScalarTensorHandleUtil(AbstractContext* ctx, in GetScalarTensorHandleUtil()165 AbstractTensorHandlePtr y; in GetScalarTensorHandleUtil()
119 AbstractTensorHandlePtr x1; in TEST_P()126 AbstractTensorHandlePtr x2; in TEST_P()167 AbstractTensorHandlePtr t; in TEST_P()232 AbstractTensorHandlePtr x; in TEST_P()
61 using AbstractTensorHandlePtr = variable
115 AbstractTensorHandlePtr x; in TEST_P()124 AbstractTensorHandlePtr y; in TEST_P()142 AbstractTensorHandlePtr x; in TEST_P()167 AbstractTensorHandlePtr A; in TEST_P()178 AbstractTensorHandlePtr B; in TEST_P()208 AbstractTensorHandlePtr x; in TEST_P()226 AbstractTensorHandlePtr x; in TEST_P()244 AbstractTensorHandlePtr x; in TEST_P()253 AbstractTensorHandlePtr y; in TEST_P()271 AbstractTensorHandlePtr x; in TEST_P()[all …]
68 AbstractTensorHandlePtr conj_output_releaser(conj_outputs[0]); in Compute()78 AbstractTensorHandlePtr exp_;97 AbstractTensorHandlePtr sqrt_;137 AbstractTensorHandlePtr A(conj_outputs[0]); in Compute()143 AbstractTensorHandlePtr B(conj_outputs[0]); in Compute()347 AbstractTensorHandlePtr Conj_X(temp_outputs[0]); in Compute()354 AbstractTensorHandlePtr Ones_X(temp_outputs[0]); in Compute()361 AbstractTensorHandlePtr Conj_XP1(temp_outputs[0]); in Compute()425 AbstractTensorHandlePtr MinusU(temp_outputs[0]); in Compute()430 AbstractTensorHandlePtr UZ(temp_outputs[0]); in Compute()
159 AbstractTensorHandlePtr X; in TEST_P()173 AbstractTensorHandlePtr Y; in TEST_P()200 AbstractTensorHandlePtr X; in TEST_P()211 AbstractTensorHandlePtr Y; in TEST_P()234 AbstractTensorHandlePtr A; in TEST_P()245 AbstractTensorHandlePtr Bias; in TEST_P()
101 AbstractTensorHandlePtr x; in TEST_P()
38 AbstractTensorHandlePtr numerical_grad; in CompareNumericalAndAutodiffGradients()
61 AbstractTensorHandlePtr owned_var_handle(var_handle); in CreateUninitializedResourceVariable()97 AbstractTensorHandlePtr owned_value(value); in ReadVariable()
100 AbstractTensorHandlePtr owned_restored_handle(restored_handle); in SingleRestore()
40 AbstractTensorHandlePtr owned_dummy_output(dummy_output); in ExecuteNoArgDummyReturnFunction()
416 AbstractTensorHandlePtr owned_resource_handle(resource_handle); in CreateAllResourceHandles()
51 using tensorflow::AbstractTensorHandlePtr;226 py::class_<AbstractTensorHandle, AbstractTensorHandlePtr>( in PYBIND11_MODULE()