| /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/libtextclassifier/native/lang_id/common/math/ |
| D | softmax.cc | 75 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/libtextclassifier/native/utils/math/ |
| D | softmax.cc | 77 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | activations.py | 47 def softmax(x, axis=-1): function
|
| D | backend.py | 4856 def softmax(x, axis=-1): function
|
| /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/lite/toco/tflite/ |
| D | op_version_test.cc | 46 std::unique_ptr<SoftmaxOperator> softmax(new SoftmaxOperator()); in TEST() local
|
| /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/core/kernels/hexagon/ |
| D | graph_transferer_test.cc | 167 Output softmax = BuildSoftmaxOps(root.WithOpName("softmax"), conv); in CreateConvGraphDef() local
188 Output softmax = BuildSoftmaxOps(root.WithOpName("softmax"), max_pool); in CreatePoolGraphDef() local
|
| /external/libtextclassifier/native/lang_id/ |
| D | lang-id.cc | 169 const std::vector<float> softmax = ComputeSoftmax(scores); in FindLanguages() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 100 auto softmax = in Compile() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 532 ShapeHandle softmax = sparse_matrix_shape_and_type.shape; in __anon5631e9dc0e02() local
|
| /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 | 1360 auto softmax = ops::Softmax(root, input); in BM_ImageNetSoftmaxFwd() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3666 def softmax(logits, axis=None, name=None, dim=None): function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 2649 Output softmax = ops::Softmax(s.WithOpName("softmax"), x); in TEST_F() local
2681 Output softmax = ops::Softmax(s.WithOpName("softmax"), floats); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 1232 auto* softmax = new SoftmaxOperator; in ConvertSoftmaxOperator() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1008 string softmax; member
|