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

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.cc 106 void* ws_data) {
113 DCHECK(ws_data != nullptr);
114 context_.ws_mem->set_data_handle(ws_data);
124 DCHECK(ws_data != nullptr);
200 T* diff_src_data, const void* ws_data) {
205 DCHECK(ws_data != nullptr);
206 context_.ws_mem->set_data_handle(const_cast<void*>(ws_data));
214 DCHECK(ws_data != nullptr);
mkl_maxpooling_op.cc 635 T* ws_data = variable
639 pooling_fwd->Execute(src_data, dst_data, ws_data);
763 void* ws_data = static_cast<void*>( variable
    [all...]
mkl_pooling_ops_common.h 82 // ws_data: output data buffer of workspace
84 void Execute(const T* src_data, T* dst_data, void* ws_data = nullptr);
224 // ws_data: input data buffer of workspace
226 const void* ws_data = nullptr);

Completed in 75 milliseconds