Searched refs:scratchpad_memory (Results 1 – 3 of 3) sorted by relevance
224 auto scratchpad_memory = make_onednn_memory( in matmul() local226 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_memory}); in matmul()
315 auto scratchpad_memory = make_onednn_memory( in deconvolution_backward_data() local319 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_memory}); in deconvolution_backward_data()
434 auto scratchpad_memory = make_onednn_memory( in convolution_backward_data() local438 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_memory}); in convolution_backward_data()