Searched refs:diff_weights_mem (Results 1 – 1 of 1) sorted by relevance
439 context_.diff_weights_mem->set_data_handle( in Execute()458 context_.diff_weights_mem->set_data_handle( in Execute()475 context_.diff_weights_mem->set_data_handle(DummyData); in Execute()497 std::shared_ptr<mkldnn::memory> diff_weights_mem; member515 diff_weights_mem(nullptr), in BatchNormBwdContext()564 context_.diff_weights_mem.reset( in Setup()576 {MKLDNN_ARG_DIFF_WEIGHTS, *context_.diff_weights_mem}}); in Setup()