Searched refs:ws_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 87 void* ws_data, in Execute() argument
96 DCHECK(ws_data != nullptr); in Execute()
97 context_.ws_mem->set_data_handle(ws_data, *fwd_stream); in Execute()
106 DCHECK(ws_data != nullptr); in Execute()
107 context_.ws_mem->set_data_handle(ws_data); in Execute()
118 DCHECK(ws_data != nullptr); in Execute()
189 T* diff_src_data, const void* ws_data, in Execute() argument
197 DCHECK(ws_data != nullptr); in Execute()
198 context_.ws_mem->set_data_handle(const_cast<void*>(ws_data), *bwd_stream); in Execute()
205 DCHECK(ws_data != nullptr); in Execute()
[all …]
|
| D | mkl_maxpooling_op.cc | 184 T* ws_data = in Compute() local
187 pooling_fwd->Execute(src_data, dst_data, ws_data, fwd_cpu_stream); in Compute()
330 void* ws_data = static_cast<void*>( in Compute() local
336 pooling_bwd->Execute(diff_dst_data, diff_src_data, ws_data, in Compute()
|
| D | mkl_pooling_ops_common.h | 86 void Execute(const T* src_data, T* dst_data, void* ws_data,
225 void Execute(const T* diff_dst_data, T* diff_src_data, const void* ws_data,
|
| D | mkl_fused_batch_norm_op.cc | 784 U* ws_data = nullptr; in Compute() local
797 ws_data = static_cast<U*>(wksp.GetOpMem().get_data_handle()); in Compute()
871 variance_op_data, fwd_cpu_stream, ws_data); in Compute()
|