Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc1286 auto diff_scale_data = diff_scale_tensor->flat<U>().data(); in Compute() local
1288 std::memcpy(reinterpret_cast<char*>(diff_scale_data), in Compute()
1355 auto diff_scale_data = (*diff_scale_tensor)->flat<U>().data(); in AllocateTFOutputs() local
1356 std::fill_n(diff_scale_data, (*diff_scale_tensor)->shape().num_elements(), in AllocateTFOutputs()