Searched defs:prepacked (Results 1 – 6 of 6) sorted by relevance
75 TORCH_LIBRARY(prepacked, m) { in TORCH_LIBRARY() argument86 TORCH_LIBRARY_IMPL(prepacked, CPU, m) { in TORCH_LIBRARY_IMPL() argument
328 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeQuantizedConv1d() local358 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeQuantizedConv2d() local388 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeQuantizedConv2dRelu() local418 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeQuantizedLinear() local448 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeQuantizedLinearRelu() local
443 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computePrepackedConv2dClampRun() local462 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computePrepackedLinearClampRun() local482 const BufHandle& prepacked = std::get<BufHandle>(inputs[1]); in computeMkldnnPrepackedConvRun() local
197 #define cl_emit_with_prepacked(cl, packet, prepacked, name) \ argument
241 #define cl_emit_with_prepacked(cl, packet, prepacked, name) \ argument
513 auto prepacked = linear_clamp_prepack_op.call( in TEST() local591 auto prepacked = conv2d_clamp_prepack_op.call( in TEST() local