Searched refs:scratchpad_m (Results 1 – 2 of 2) sorted by relevance
242 auto scratchpad_m = make_onednn_memory( in convolution() local244 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_m}); in convolution()343 auto scratchpad_m = make_onednn_memory( in convolution_backward_weights() local345 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_m}); in convolution_backward_weights()
229 auto scratchpad_m = make_onednn_memory( in deconvolution() local231 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_m}); in deconvolution()419 auto scratchpad_m = make_onednn_memory( in deconvolution_backward_weights() local421 args.insert({DNNL_ARG_SCRATCHPAD, scratchpad_m}); in deconvolution_backward_weights()