Searched refs:in_grads (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op.cc | 396 const Tensor& in_grads, const Tensor& in_image, in launch()
398 const int64 batch = in_grads.dim_size(0); in launch()
399 const int64 rows = in_grads.dim_size(1); in launch()
400 const int64 cols = in_grads.dim_size(2); in launch()
401 const int64 depth = in_grads.dim_size(3); in launch()
403 auto grads_shaped = in_grads.shaped<T, 2>({nodes * batch, depth}); in launch()
475 const Tensor& in_grads, const Tensor& in_image, in launch()
490 const int64 batch = in_grads.dim_size(0); in launch()
491 const int64 rows = in_grads.dim_size(1); in launch()
492 const int64 cols = in_grads.dim_size(2); in launch()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 318 in_grads = functional_ops.symbolic_gradient(input=f_in, Tout=f_types, f=f)
319 return in_grads
683 in_grads = _MaybeCompile(grad_scope, op, func_call,
688 in_grads = _MaybeCompile(grad_scope, op, func_call,
690 in_grads = _AsList(in_grads)
691 _VerifyGeneratedGradients(in_grads, op)
692 if gate_gradients and len([x for x in in_grads
699 in_grads = control_flow_ops.tuple(in_grads)
700 _LogOpGradients(op, out_grads, in_grads)
704 in_grads = [None] * len(_Inputs(op, xs_set))
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lrn_ops.cc | 114 xla::XlaOp in_grads = ctx->Input(0); in Compile() local
155 in_grads); in Compile()
167 xla::Mul(in_grads, in Compile()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 334 const std::vector<Gradient*>& in_grads, absl::Span<Gradient*> out_grads);
928 const std::vector<Gradient*>& in_grads, absl::Span<Gradient*> out_grads) { in ForwardpropFromTape() argument
972 std::vector<Gradient*> grad(in_grads.size()); in ForwardpropFromTape()
1007 Gradient* in_grad = in_grads[grad_index]; in ForwardpropFromTape()
1058 std::vector<Gradient*> in_grads; in Accumulate() local
1059 in_grads.reserve(input_tensors.size()); in Accumulate()
1070 in_grads.push_back(zero); in Accumulate()
1072 in_grads.push_back(nullptr); in Accumulate()
1075 in_grads.push_back(current_grad->second); in Accumulate()
1091 backward_function_deleter, in_grads, absl::MakeSpan(forward_grads))); in Accumulate()
[all …]
|