Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc459 auto coef0_one_by_depth = in operator ()() local
461 auto coef0_rest_by_depth = coef0_one_by_depth.broadcast(bcast_spec); in operator ()()
496 auto coef1 = (scale.reshape(one_by_depth) * coef0_one_by_depth) in operator ()()
498 auto coef2 = (coef0_one_by_depth.square() * y_backprop_centered_mean) in operator ()()