Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc1211 U* weights_data_tf = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute() local
1214 weights_data_tf[k] = scale_tf[k]; in Compute()
1215 weights_data_tf[k + depth_] = static_cast<U>(0); in Compute()
1264 U* weights_data = weights_data_tf; in Compute()