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

  /external/tensorflow/tensorflow/core/kernels/
mkl_avgpooling_op.cc 97 void* workspace_buf; variable
99 &workspace_buf);
142 mkl_context.pooling_res[dnnResourceWorkspace] = workspace_buf;
307 void* workspace_buf; variable
309 &workspace_buf);
310 mkl_context.pooling_res[dnnResourceWorkspace] = workspace_buf;
mkl_maxpooling_op.cc 122 void* workspace_buf = nullptr; variable
397 void* workspace_buf = nullptr; local
428 &workspace_buf);
429 pooling_resfwd[dnnResourceWorkspace] = workspace_buf;
    [all...]

Completed in 125 milliseconds