Searched refs:in_grads (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lrn_op.cc | 318 const Tensor& in_grads, const Tensor& in_image, in launch()
320 const int64 batch = in_grads.dim_size(0); in launch()
321 const int64 rows = in_grads.dim_size(1); in launch()
322 const int64 cols = in_grads.dim_size(2); in launch()
323 const int64 depth = in_grads.dim_size(3); in launch()
325 auto grads_shaped = in_grads.shaped<T, 2>({nodes * batch, depth}); in launch()
392 const Tensor& in_grads, const Tensor& in_image, in launch()
406 const int64 batch = in_grads.dim_size(0); in launch()
407 const int64 rows = in_grads.dim_size(1); in launch()
408 const int64 cols = in_grads.dim_size(2); in launch()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 381 in_grads = functional_ops.symbolic_gradient(
385 return in_grads
730 in_grads = _MaybeCompile(grad_scope, op, func_call,
735 in_grads = _MaybeCompile(grad_scope, op, func_call,
737 in_grads = _AsList(in_grads)
738 _VerifyGeneratedGradients(in_grads, op)
739 if gate_gradients and len([x for x in in_grads
746 in_grads = control_flow_ops.tuple(in_grads)
747 _LogOpGradients(op, out_grads, in_grads)
751 in_grads = [None] * len(_Inputs(op, xs))
[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()
|