Searched refs:a_mem (Results 1 – 1 of 1) sorted by relevance
544 context_.a_mem->set_data_handle(static_cast<void*>(const_cast<T*>(a_data)), in Execute()551 context_.a_mem->set_data_handle(static_cast<void*>(const_cast<T*>(a_data))); in Execute()558 context_.a_mem->set_data_handle(DummyData); in Execute()567 std::shared_ptr<mkldnn::memory> a_mem; member585 : a_mem(nullptr), in MklMatMulContext()615 context_.a_mem.reset( in Setup()624 context_.net_args.push_back({{MKLDNN_ARG_SRC, *context_.a_mem}, in Setup()