Home
last modified time | relevance | path

Searched refs:create_batchnorm_context (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBatchnorm.h49 c10::intrusive_ptr<BatchNormPackedContext> create_batchnorm_context(
H A DRegister.cpp296 TORCH_FN(create_batchnorm_context)); in TORCH_LIBRARY_IMPL()
H A DBatchnorm.cpp183 c10::intrusive_ptr<BatchNormPackedContext> create_batchnorm_context( in create_batchnorm_context() function