| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_2/ |
| D | log_softmax.example.cpp | 7 namespace generated_tests::log_softmax { namespace
74 namespace generated_tests::log_softmax { namespace
141 namespace generated_tests::log_softmax { namespace
208 namespace generated_tests::log_softmax { namespace
275 namespace generated_tests::log_softmax { namespace
342 namespace generated_tests::log_softmax { namespace
409 namespace generated_tests::log_softmax { namespace
476 namespace generated_tests::log_softmax { namespace
543 namespace generated_tests::log_softmax { namespace
610 namespace generated_tests::log_softmax { namespace
[all …]
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Softmax.cpp | 37 bool log_softmax) { in add_softmax_node()
113 void log_softmax(ComputeGraph& graph, const std::vector<ValueRef>& args) { in log_softmax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | softmax_spmd_expander.cc | 159 auto log_softmax = builder.create<mlir::TF::SubOp>( in ComputeLogSoftmax() local
169 bool log_softmax) { in ComputeShardedSoftmax()
579 const mlir::Value log_softmax = in ExpandOp() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | activation.h | 307 inline Tensor log_softmax( in log_softmax() function
336 inline Tensor log_softmax( in log_softmax() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | SoftMax.cpp | 499 Tensor log_softmax(const Tensor& input_, const int64_t dim_, std::optional<ScalarType> dtype) { in log_softmax() function
568 Tensor log_softmax(const Tensor& self, Dimname dim, std::optional<ScalarType> dtype) { in log_softmax() function
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | LogSoftmax.cpp | 24 namespace log_softmax { namespace
|
| /aosp_15_r20/external/executorch/backends/arm/_passes/ |
| H A D | decompose_softmaxes_pass.py | 22 log_softmax = (torch.ops.aten.log_softmax.int, exir_ops.edge.aten._log_softmax.default) variable
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LogSoftmax.cpp | 31 namespace log_softmax { namespace
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | activations.py | 515 log_softmax = nn.log_softmax variable
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | __init__.py | 321 log_softmax = _add_docstr( variable
|
| /aosp_15_r20/external/pytorch/torch/_refs/special/ |
| H A D | __init__.py | 206 def log_softmax( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Softmax.cpp | 192 Tensor log_softmax( in log_softmax() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
| H A D | softmax.cpp | 11 bool log_softmax) { in computeSoftmax()
|
| /aosp_15_r20/external/pytorch/torch/special/ |
| H A D | __init__.py | 695 log_softmax = _add_docstr(_special.special_log_softmax, variable
|
| /aosp_15_r20/external/pytorch/torch/_refs/nn/functional/ |
| H A D | __init__.py | 623 def log_softmax( function
|
| /aosp_15_r20/external/pytorch/torch/masked/ |
| H A D | _ops.py | 1733 def log_softmax( function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset13.py | 40 def log_softmax(g: jit_utils.GraphContext, input, dim, dtype=None): function
|
| H A D | symbolic_opset9.py | 2207 def log_softmax(g: jit_utils.GraphContext, input, dim, dtype=None): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/ |
| H A D | special.h | 596 inline Tensor log_softmax( in log_softmax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_ops.py | 3918 def log_softmax(logits, axis=None, name=None, dim=None): function
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 2220 def log_softmax( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_api_test.cpp | 4525 void test_softmax(const at::IntArrayRef shape, bool log_softmax = false) { in test_softmax()
4561 bool log_softmax = false; in TEST_F() local
4578 bool log_softmax = true; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 783 def log_softmax( function
|