Home
last modified time | relevance | path

Searched refs:grad_outputs (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc44 std::vector<Output>* grad_outputs) { in PackGrad() argument
50 grad_outputs->reserve(N); in PackGrad()
53 grad_outputs->emplace_back(o); in PackGrad()
61 std::vector<Output>* grad_outputs) { in UnpackGrad() argument
64 grad_outputs->push_back(Stack(scope, grad_inputs, Stack::Axis(axis))); in UnpackGrad()
71 std::vector<Output>* grad_outputs) { in IdentityGrad() argument
72 grad_outputs->push_back(Identity(scope, grad_inputs[0])); in IdentityGrad()
79 std::vector<Output>* grad_outputs) { in RefIdentityGrad() argument
80 grad_outputs->push_back(Identity(scope, grad_inputs[0])); in RefIdentityGrad()
87 std::vector<Output>* grad_outputs) { in QuantizeAndDequantizeGrad() argument
[all …]
Dmath_grad.cc58 std::vector<Output>* grad_outputs) { in AbsGrad() argument
60 grad_outputs->push_back(Mul(scope, grad_inputs[0], Sign(scope, op.input(0)))); in AbsGrad()
67 std::vector<Output>* grad_outputs) { in NegGrad() argument
69 grad_outputs->push_back(Neg(scope, grad_inputs[0])); in NegGrad()
76 std::vector<Output>* grad_outputs) { in InvGrad() argument
78 grad_outputs->push_back( in InvGrad()
87 std::vector<Output>* grad_outputs) { in SquareGrad() argument
92 grad_outputs->push_back( in SquareGrad()
100 std::vector<Output>* grad_outputs) { in SqrtGrad() argument
102 grad_outputs->push_back( in SqrtGrad()
[all …]
Dnn_grad.cc29 std::vector<Output>* grad_outputs) { in SoftmaxGrad() argument
45 grad_outputs->push_back(dx); in SoftmaxGrad()
79 std::vector<Output>* grad_outputs) { in SoftmaxCrossEntropyWithLogitsGrad() argument
109 grad_outputs->push_back(grad); in SoftmaxCrossEntropyWithLogitsGrad()
110 grad_outputs->push_back(BroadcastMul(scope, grad_loss, minus_log_softmax)); in SoftmaxCrossEntropyWithLogitsGrad()
118 std::vector<Output>* grad_outputs) { in LogSoftmaxGrad() argument
123 grad_outputs->push_back(dx); in LogSoftmaxGrad()
130 std::vector<Output>* grad_outputs) { in ReluGradHelper() argument
132 grad_outputs->push_back(dx); in ReluGradHelper()
139 std::vector<Output>* grad_outputs) { in Relu6GradHelper() argument
[all …]
Dimage_grad.cc28 std::vector<Output>* grad_outputs) { in ResizeNearestNeighborGradHelper() argument
39 grad_outputs->push_back(internal::ResizeNearestNeighborGrad( in ResizeNearestNeighborGradHelper()
43 grad_outputs->push_back(NoGradient()); in ResizeNearestNeighborGradHelper()
50 std::vector<Output>* grad_outputs) { in ResizeBilinearGradHelper() argument
57 grad_outputs->push_back(internal::ResizeBilinearGrad( in ResizeBilinearGradHelper()
61 grad_outputs->push_back(NoGradient()); in ResizeBilinearGradHelper()
68 std::vector<Output>* grad_outputs) { in ResizeBicubicGradHelper() argument
76 grad_outputs->push_back(internal::ResizeBicubicGrad( in ResizeBicubicGradHelper()
80 grad_outputs->push_back(NoGradient()); in ResizeBicubicGradHelper()
87 std::vector<Output>* grad_outputs) { in ScaleAndTranslateGradHelper() argument
[all …]
Ddata_flow_grad.cc46 std::vector<Output>* grad_outputs) { in DynamicPartitionGrad() argument
102 grad_outputs->push_back(Reshape(scope, reconstructed, Shape(scope, data))); in DynamicPartitionGrad()
104 grad_outputs->push_back(NoGradient()); in DynamicPartitionGrad()
111 std::vector<Output>* grad_outputs) { in DynamicStitchGrad() argument
125 grad_outputs->push_back(NoGradient()); in DynamicStitchGrad()
145 grad_outputs->push_back(Gather(scope, grad_inputs[0], index)); in DynamicStitchGrad()
Dgrad_testutil.cc25 std::vector<Output>* grad_outputs) { in CallGradFunction() argument
29 TF_RETURN_IF_ERROR(grad_fn(scope, op, grad_inputs, grad_outputs)); in CallGradFunction()
Dgrad_testutil.h30 std::vector<Output>* grad_outputs);
DREADME.md18 std::vector<Output>* grad_outputs) {
Dmath_grad_test.cc873 std::vector<Output> grad_outputs; in TEST_F() local
874 TF_EXPECT_OK(AddSymbolicGradients(scope_, {y}, {x}, &grad_outputs)); in TEST_F()
878 session.Run({{x, {-3.0f, 0.0f, 3.0f}}}, grad_outputs, &grad_result)); in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dgradients_test.cc105 std::vector<Output> grad_outputs; in TEST_F() local
107 AddSymbolicGradients(scope, {z}, {x, y}, {dz}, &grad_outputs)); in TEST_F()
132 std::vector<Output> grad_outputs; in TEST_F() local
133 TF_ASSERT_OK(AddSymbolicGradients(scope, {z}, {x, y}, &grad_outputs)); in TEST_F()
164 std::vector<Output> grad_outputs; in TEST_F() local
166 AddSymbolicGradients(scope, {z}, {u, v}, {dz}, &grad_outputs)); in TEST_F()
201 std::vector<Output> grad_outputs; in TEST_F() local
203 &grad_outputs)); in TEST_F()
232 std::vector<Output> grad_outputs; in TEST_F() local
234 {dx, dy, dz}, &grad_outputs)); in TEST_F()
[all …]
Dgradients.cc55 std::vector<Output>* grad_outputs);
81 std::vector<Output>* grad_outputs);
143 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) in SymbolicGradientBuilder() argument
149 grad_outputs_(grad_outputs) {} in SymbolicGradientBuilder()
382 std::vector<Output>* grad_outputs) { in CallGradFunction() argument
385 TF_RETURN_IF_ERROR(grad_fn(scope_, op, grad_inputs, grad_outputs)); in CallGradFunction()
561 std::vector<Output>* grad_outputs) { in AddSymbolicGradients() argument
563 inputs, grad_inputs, grad_outputs); in AddSymbolicGradients()
570 std::vector<Output>* grad_outputs) { in AddSymbolicGradients() argument
577 grad_outputs); in AddSymbolicGradients()
Dwhile_gradients.cc140 std::vector<Output>* grad_outputs) { in AddWhileGradientLoop() argument
172 frame_name, grad_outputs, in AddWhileGradientLoop()
181 std::vector<Output>* grad_outputs) { in AddWhileLoopGradient() argument
195 scope, grad_outputs); in AddWhileLoopGradient()
Dgradients.h34 std::vector<Output>* grad_outputs);
41 std::vector<Output>* grad_outputs);
Dwhile_gradients.h36 std::vector<Output>* grad_outputs);
Dgrad_op_registry.h32 std::vector<Output>* grad_outputs);
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1534 TF_Output grad_outputs[2]; in TestGradientsSuccess() local
1541 grad_outputs); in TestGradientsSuccess()
1552 RunGraphsAndCompareOutputs(grad_outputs, expected_grad_outputs); in TestGradientsSuccess()
1558 TF_Output grad_outputs[1]; in TestGradientsError() local
1563 grad_outputs); in TestGradientsError()
1574 void RunGraphsAndCompareOutputs(TF_Output* grad_outputs, in RunGraphsAndCompareOutputs() argument
1581 grad_outputs_vec.assign(grad_outputs, grad_outputs + 2); in RunGraphsAndCompareOutputs()
1609 int noutputs, TF_Output* grad_outputs) { in AddGradients() argument
1617 ninputs, grad_inputs, s_, grad_outputs); in AddGradients()
1620 ninputs, nullptr, s_, grad_outputs); in AddGradients()
[all …]