Searched refs:add_softmax_node (Results 1 – 6 of 6) sorted by relevance
| /external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| D | Softmax.cpp | 32 void add_softmax_node( in add_softmax_node() function
109 return add_softmax_node( in softmax()
115 return add_softmax_node( in log_softmax()
|
| D | Softmax.h | 19 void add_softmax_node(
|
| D | SDPA.cpp | 318 add_softmax_node(graph, attn_weight, width, attn_weight_softmax, false); in sdpa_with_kv_cache_impl()
|
| /external/ComputeLibrary/arm_compute/graph/ |
| D | GraphBuilder.h | 530 … static NodeID add_softmax_node(Graph &g, NodeParams params, NodeIdxPair input, float beta = 1.f);
|
| /external/ComputeLibrary/src/graph/ |
| D | GraphBuilder.cpp | 710 NodeID GraphBuilder::add_softmax_node(Graph &g, NodeParams params, NodeIdxPair input, float beta) in add_softmax_node() function in arm_compute::graph::GraphBuilder
|
| /external/ComputeLibrary/arm_compute/graph/frontend/ |
| D | Layers.h | 1339 return GraphBuilder::add_softmax_node(s.graph(), common_params, input, _beta); in create_layer()
|