HomeSort by relevance Sort by last modified time
    Searched refs:workspace_pd (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 273 memory::primitive_desc workspace_pd = local
275 size_t workspace_bytes = workspace_pd.get_size();
285 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
512 memory::primitive_desc workspace_pd,
516 workspace_dnn_data->SetUsrMem(workspace_pd, &workspace_tensor);
mkl_maxpooling_op.cc 660 memory::primitive_desc workspace_pd = local
662 size_t workspace_bytes = workspace_pd.get_size();
671 dnn_data_wksp->SetUsrMem(workspace_pd, workspace_tensor);
    [all...]

Completed in 581 milliseconds