Home
last modified time | relevance | path

Searched defs:variantPack (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v8.cpp394 auto variantPack = in run_conv_plan() local
419 auto variantPack = in run_conv_plan_fused() local
723 auto variantPack = in get_plans_from_find() local
778 auto variantPack = in get_plans_from_find_fused() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DBinaryOps.cpp153 auto variantPack = cudnn_frontend::VariantPackBuilder() in add() local
H A DConv.cpp130 auto variantPack = cudnn_frontend::VariantPackBuilder() in apply_impl_helper() local
H A DLinear.cpp158 auto variantPack = cudnn_frontend::VariantPackBuilder() in apply_impl_helper() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc4477 auto variantPack = in operator ()() local