Searched refs:weights_mem (Results 1 – 1 of 1) sorted by relevance
89 context_.weights_mem->set_data_handle( in Execute()108 context_.weights_mem->set_data_handle( in Execute()128 context_.weights_mem->set_data_handle(DummyData); in Execute()158 std::shared_ptr<mkldnn::memory> weights_mem; member177 weights_mem(nullptr), in BatchNormFwdContext()214 context_.weights_mem.reset( in Setup()240 {MKLDNN_ARG_WEIGHTS, *context_.weights_mem}, in Setup()254 {MKLDNN_ARG_WEIGHTS, *context_.weights_mem}, in Setup()262 {MKLDNN_ARG_WEIGHTS, *context_.weights_mem}, in Setup()287 {MKLDNN_ARG_WEIGHTS, *context_.weights_mem}, in Setup()[all …]