Home
last modified time | relevance | path

Searched refs:in_grads (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlrn_op.cc314 const Tensor& in_grads, const Tensor& in_image, in launch()
316 const int64 batch = in_grads.dim_size(0); in launch()
317 const int64 rows = in_grads.dim_size(1); in launch()
318 const int64 cols = in_grads.dim_size(2); in launch()
319 const int64 depth = in_grads.dim_size(3); in launch()
321 auto grads_shaped = in_grads.shaped<T, 2>({nodes * batch, depth}); in launch()
388 const Tensor& in_grads, const Tensor& in_image, in launch()
402 const int64 batch = in_grads.dim_size(0); in launch()
403 const int64 rows = in_grads.dim_size(1); in launch()
404 const int64 cols = in_grads.dim_size(2); in launch()
[all …]
Dmkl_lrn_op.cc370 const Tensor& in_grads = MklGetInput(context, 0); in Compute() local
422 context, in_grads.dims() == 4, in Compute()
536 const Tensor& in_grads = MklGetInput(context, 0); in MklPrepareLRNGradInput() local
544 static_cast<const void*>(in_grads.flat<T>().data())); in MklPrepareLRNGradInput()
629 Tensor in_grads; in MklDefaultToEigen() local
638 in_grads = in MklDefaultToEigen()
641 in_grads = MklGetInput(context, 0); in MklDefaultToEigen()
658 const int64 batch = static_cast<int64>(in_grads.dim_size(0)); in MklDefaultToEigen()
659 const int64 rows = static_cast<int64>(in_grads.dim_size(1)); in MklDefaultToEigen()
660 const int64 cols = static_cast<int64>(in_grads.dim_size(2)); in MklDefaultToEigen()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py357 in_grads = functional_ops._symbolic_gradient(input=f_in, Tout=f_types, f=f)
359 return in_grads
610 in_grads = _MaybeCompile(grad_scope, op, func_call,
615 in_grads = _MaybeCompile(grad_scope, op, func_call,
617 in_grads = _AsList(in_grads)
618 _VerifyGeneratedGradients(in_grads, op)
619 if gate_gradients and len([x for x in in_grads
623 in_grads = control_flow_ops.tuple(in_grads)
624 _LogOpGradients(op, out_grads, in_grads)
628 in_grads = [None] * len(op.inputs)
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlrn_ops.cc110 xla::ComputationDataHandle in_grads = ctx->Input(0); in Compile() local
147 in_grads); in Compile()
157 builder->Mul(in_grads, in Compile()