Home
last modified time | relevance | path

Searched refs:grad_inputs (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
Darray_grad.cc43 const std::vector<Output>& grad_inputs, in PackGrad() argument
51 auto grad_op = Unstack(scope, grad_inputs[0], N, Unstack::Axis(axis)); in PackGrad()
60 const std::vector<Output>& grad_inputs, in UnpackGrad() argument
64 grad_outputs->push_back(Stack(scope, grad_inputs, Stack::Axis(axis))); in UnpackGrad()
70 const std::vector<Output>& grad_inputs, in IdentityGrad() argument
72 grad_outputs->push_back(Identity(scope, grad_inputs[0])); in IdentityGrad()
78 const std::vector<Output>& grad_inputs, in RefIdentityGrad() argument
80 grad_outputs->push_back(Identity(scope, grad_inputs[0])); in RefIdentityGrad()
86 const std::vector<Output>& grad_inputs, in QuantizeAndDequantizeGrad() argument
88 grad_outputs->push_back(Identity(scope, grad_inputs[0])); in QuantizeAndDequantizeGrad()
[all …]
Dmath_grad.cc57 const std::vector<Output>& grad_inputs, in AbsGrad() argument
60 grad_outputs->push_back(Mul(scope, grad_inputs[0], Sign(scope, op.input(0)))); in AbsGrad()
66 const std::vector<Output>& grad_inputs, in NegGrad() argument
69 grad_outputs->push_back(Neg(scope, grad_inputs[0])); in NegGrad()
75 const std::vector<Output>& grad_inputs, in InvGrad() argument
79 internal::ReciprocalGrad(scope, op.output(0), grad_inputs[0])); in InvGrad()
86 const std::vector<Output>& grad_inputs, in SquareGrad() argument
93 Mul(scope, grad_inputs[0], ConjugateHelper(scope, dydx))); in SquareGrad()
99 const std::vector<Output>& grad_inputs, in SqrtGrad() argument
103 internal::SqrtGrad(scope, op.output(0), grad_inputs[0])); in SqrtGrad()
[all …]
Dnn_grad.cc28 const std::vector<Output>& grad_inputs, in SoftmaxGrad() argument
41 auto dyy = Mul(scope, grad_inputs[0], y); in SoftmaxGrad()
43 auto sub = Sub(scope, grad_inputs[0], sum); in SoftmaxGrad()
78 const std::vector<Output>& grad_inputs, in SoftmaxCrossEntropyWithLogitsGrad() argument
90 auto grad_loss = grad_inputs[0]; in SoftmaxCrossEntropyWithLogitsGrad()
91 auto grad_grad = grad_inputs[1]; in SoftmaxCrossEntropyWithLogitsGrad()
117 const std::vector<Output>& grad_inputs, in LogSoftmaxGrad() argument
120 auto sum = Sum(scope, grad_inputs[0], {1}, Sum::KeepDims(true)); in LogSoftmaxGrad()
122 auto dx = Sub(scope, grad_inputs[0], mul); in LogSoftmaxGrad()
129 const std::vector<Output>& grad_inputs, in ReluGradHelper() argument
[all …]
Dimage_grad.cc27 const std::vector<Output>& grad_inputs, in ResizeNearestNeighborGradHelper() argument
40 scope, grad_inputs[0], x_shape, in ResizeNearestNeighborGradHelper()
49 const std::vector<Output>& grad_inputs, in ResizeBilinearGradHelper() argument
58 scope, grad_inputs[0], op.input(0), in ResizeBilinearGradHelper()
67 const std::vector<Output>& grad_inputs, in ResizeBicubicGradHelper() argument
77 scope, grad_inputs[0], op.input(0), in ResizeBicubicGradHelper()
86 const std::vector<Output>& grad_inputs, in ScaleAndTranslateGradHelper() argument
94 scope, grad_inputs[0], op.input(0), op.input(2), op.input(3), in ScaleAndTranslateGradHelper()
107 const std::vector<Output>& grad_inputs, in CropAndResizeGradHelper() argument
115 scope, grad_inputs[0], op.input(1), op.input(2), image_shape, input_type, in CropAndResizeGradHelper()
[all …]
Ddata_flow_grad.cc45 const std::vector<Output>& grad_inputs, in DynamicPartitionGrad() argument
99 DynamicStitch(scope, partitioned_indices.outputs, grad_inputs); in DynamicPartitionGrad()
110 const std::vector<Output>& grad_inputs, in DynamicStitchGrad() argument
145 grad_outputs->push_back(Gather(scope, grad_inputs[0], index)); in DynamicStitchGrad()
Dgrad_testutil.cc24 const std::vector<Output>& grad_inputs, in CallGradFunction() argument
29 TF_RETURN_IF_ERROR(grad_fn(scope, op, grad_inputs, grad_outputs)); in CallGradFunction()
Dgrad_testutil.h29 const std::vector<Output>& grad_inputs,
DREADME.md17 const std::vector<Output>& grad_inputs,
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc54 const std::vector<Output>& grad_inputs,
80 const std::vector<Output>& grad_inputs,
143 const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) in SymbolicGradientBuilder() argument
148 grad_inputs_(grad_inputs), in SymbolicGradientBuilder()
381 const std::vector<Output>& grad_inputs, in CallGradFunction() argument
385 TF_RETURN_IF_ERROR(grad_fn(scope_, op, grad_inputs, grad_outputs)); in CallGradFunction()
560 const std::vector<Output>& grad_inputs, in AddSymbolicGradients() argument
563 inputs, grad_inputs, grad_outputs); in AddSymbolicGradients()
571 std::vector<Output> grad_inputs; in AddSymbolicGradients() local
572 grad_inputs.reserve(outputs.size()); in AddSymbolicGradients()
[all …]
Dwhile_gradients.cc137 const std::vector<Output>& grad_inputs, in AddWhileGradientLoop() argument
141 DCHECK_EQ(grad_inputs.size(), while_ctx->body_outputs().size()); in AddWhileGradientLoop()
171 TF_RETURN_IF_ERROR(BuildWhileLoop(scope, grad_inputs, cond_fn, body_fn, in AddWhileGradientLoop()
180 const std::vector<Output>& grad_inputs, in AddWhileLoopGradient() argument
194 return AddWhileGradientLoop(while_ctx, grad_inputs, backprop_counter_cond, in AddWhileLoopGradient()
Dwhile_gradients.h35 const std::vector<Output>& grad_inputs,
Dgradients.h33 const std::vector<Output>& grad_inputs,
Dgrad_op_registry.h31 const std::vector<Output>& grad_inputs,
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Drev_block_lib.py572 grad_inputs = grads[:len(inputs)]
574 return grad_inputs, grad_vars
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1611 TF_Output grad_inputs[1]; in AddGradients() local
1615 grad_inputs[0] = TF_Output{grad_inputs_op, 0}; in AddGradients()
1617 ninputs, grad_inputs, s_, grad_outputs); in AddGradients()