Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DNativeLayerNorm.cpp55 std::tuple<Tensor, Tensor, Tensor> native_layer_norm( in native_layer_norm() function
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DNativeLayerNorm.cpp124 void native_layer_norm(ComputeGraph& graph, const std::vector<ValueRef>& args) { in native_layer_norm() function
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py1122 def native_layer_norm( function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py3190 def native_layer_norm( function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py2684 def native_layer_norm( function