Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:grads_shaped (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
lrn_op.cc	321 auto grads_shaped = in_grads.shaped<T, 2>({nodes * batch, depth}); local 328 auto shard = [this, activations, in_shaped, grads_shaped, out_shaped, 363 dyi *= grads_shaped(i, j);
mkl_lrn_op.cc	664 auto grads_shaped = in_grads.shaped<T, 2>({nodes * batch, depth}); local 678 auto shard = [this, activations, in_shaped, grads_shaped, out_shaped, 697 dyi *= grads_shaped(i, j);     [all...]

Completed in 452 milliseconds