Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc882 auto batch_variance_data = batch_variance_tensor->flat<U>().data(); in Compute() local
889 batch_variance_data[k] = in Compute()
897 batch_variance_data[k] = one_minus_factor * est_variance_data[k] + in Compute()
905 std::memcpy(batch_variance_data, variance_data, depth_ * sizeof(U)); in Compute()
1004 auto batch_variance_data = (*batch_variance_tensor)->flat<U>().data(); in AllocateTFOutputs() local
1005 std::fill_n(batch_variance_data, num_elements, static_cast<U>(NAN)); in AllocateTFOutputs()