Searched refs:diff_weights (Results 1 – 1 of 1) sorted by relevance
1159 MklDnnData<U> diff_weights(&cpu_engine_); in Compute() local1218 diff_weights.AllocateBuffer(2 * depth_ * sizeof(U)); in Compute()1266 U* diff_weights_data = static_cast<U*>(diff_weights.GetAllocatedBuffer()); in Compute()