| 494 | } |
| 495 | |
| 496 | void ConfigureWorkspace(const Tensor& workspace_tensor, |
| 497 | MEMORY_PRIMITIVE_DESC workspace_pd, |
| 498 | MklDnnData<uint8>* workspace_dnn_data) { |
| 499 | DCHECK(workspace_dnn_data); |
| 500 | |
| 501 | workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor); |
| 502 | } |
| 503 | |
| 504 | // Fallback implementation - Taken from lrn_op.cc |
| 505 | // TODO(intel-tf) Check if we can use EigenLRNOp directly |