| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 348 T* softmax) { in CalculateRowSoftmax()
372 const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel2D()
397 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D()
481 const T* softmax, const int grad_softmax_begin, const int grad_softmax_end, in CalculateRowSoftmaxGrad()
547 const T* softmax, const int* grad_softmax_row_ptr, in CSRSparseMatrixSoftmaxGradKernel2D()
567 const int* softmax_row_ptr, const int* softmax_col_ind, const T* softmax, in CSRSparseMatrixSoftmaxGradKernel3D()
608 OpKernelContext* ctx, const CSRSparseMatrix& softmax, in CSRSparseMatrixSoftmaxGradGPUImpl()
|
| D | softmax_op.cc | 76 functor::CSRSparseMatrixSoftmax<Device, T> softmax; in Compute() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | tf_softmax_test.py | 25 def softmax(x): function
|
| /external/libtextclassifier/native/lang_id/common/math/ |
| D | softmax.cc | 76 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/libtextclassifier/native/utils/math/ |
| D | softmax.cc | 77 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/ComputeLibrary/examples/ |
| D | neon_copy_objects.cpp | 156 NESoftmaxLayer softmax{}; member in NEONCopyObjectsExample
|
| D | neon_cnn.cpp | 271 std::unique_ptr<NESoftmaxLayer> softmax{}; member in NEONCNNExample
|
| /external/armnn/src/backends/cl/test/ |
| D | ClContextSerializerTests.cpp | 23 armnn::IConnectableLayer* softmax = net->AddSoftmaxLayer(armnn::SoftmaxDescriptor(), "softmax"); in CreateNetwork() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 81 auto softmax = ops::Softmax(s.WithOpName("softmax"), logits); in CreateMiniGraph() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | activations.py | 41 def softmax(x, axis=-1): function
|
| D | backend.py | 4783 def softmax(x, axis=-1): function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | softmax_op_benchmark.cc | 47 OutT softmax) { in ComputeSoftmax()
|
| /external/tensorflow/tensorflow/tools/android/test/src/org/tensorflow/demo/ |
| D | TensorFlowYoloDetector.java | 118 private void softmax(final float[] vals) { in softmax() method in TensorFlowYoloDetector
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 146 auto softmax = builder.create<mlir::TF::DivOp>( in ComputeSoftmax() local
581 const mlir::Value softmax = in ExpandOp() local
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | JsonPrinterTestImpl.cpp | 142 IConnectableLayer* softmax = net->AddSoftmaxLayer(softmaxDescriptor, "softmax"); in GetSoftmaxProfilerJson() local
|
| /external/libtextclassifier/native/lang_id/ |
| D | lang-id.cc | 170 const std::vector<float> softmax = ComputeSoftmax(scores); in FindLanguages() local
|
| /external/armnn/src/backends/reference/test/ |
| D | RefEndToEndTests.cpp | 120 IConnectableLayer* softmax = net->AddSoftmaxLayer(SoftmaxDescriptor(), "softmax"); variable
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 532 ShapeHandle softmax = sparse_matrix_shape_and_type.shape; in __anon6adce4610e02() local
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonTensorHandleTests.cpp | 36 IConnectableLayer* softmax = network->AddSoftmaxLayer(descriptor); variable
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 1032 def softmax(logits: ragged_tensor.Ragged, axis=None, name=None): function
|
| /external/armnn/src/armnn/test/ |
| D | RuntimeTests.cpp | 524 armnn::IConnectableLayer* softmax = net->AddSoftmaxLayer(armnn::SoftmaxDescriptor(), "softmax"); variable
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 119 auto softmax = Exp(scope, op.output(0)); in LogSoftmaxGrad() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nn_ops_test.cc | 1389 auto softmax = ops::Softmax(root, input); in BM_ImageNetSoftmaxFwd() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3904 def softmax(logits, axis=None, name=None, dim=None): function
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 1239 auto* softmax = new SoftmaxOperator; in ConvertSoftmaxOperator() local
|