Home
last modified time | relevance | path

Searched refs:weight_mem (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_ops_common.h107 context_.weight_mem->set_data_handle( in Execute()
116 context_.weight_mem->set_data_handle( in Execute()
127 context_.weight_mem->set_data_handle(DummyData); in Execute()
142 std::shared_ptr<mkldnn::memory> weight_mem; member
164 weight_mem(nullptr), in MklDnnMatMulFwdContext()
267 context_.weight_mem.reset(new memory(context_.fwd_pd.get()->weights_desc(), in Setup()
279 {MKLDNN_ARG_WEIGHTS, *context_.weight_mem}, in Setup()