Home
last modified time | relevance | path

Searched refs:get_optional_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBatchnorm.cpp176 get_optional_tensor(unpacked, ListArgs::kWeight), in pack()
177 get_optional_tensor(unpacked, ListArgs::kBias), in pack()
178 get_optional_tensor(unpacked, ListArgs::kRunningMean), in pack()
179 get_optional_tensor(unpacked, ListArgs::kRunningVar), in pack()
H A DLayernorm.cpp45 get_optional_tensor(unpacked, ListArgs::kWeight), in pack()
46 get_optional_tensor(unpacked, ListArgs::kBias), in pack()
H A DCommon.h79 inline std::optional<Tensor> get_optional_tensor( in get_optional_tensor() function
H A DConvolution.cpp1002 get_optional_tensor(unpacked, Unpacked::Bias), in pack()
1290 get_optional_tensor(unpacked_, Conv2dPackedContext::Unpacked::Bias), in unpack()
1360 get_optional_tensor(unpacked, Unpacked::Bias), in pack()
H A DMm.cpp972 get_optional_tensor(unpacked, Unpacked::Bias)); in pack()