| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | nn.py | 35 softmax = core.define_unary_op('softmax', nn.softmax) variable
|
| /external/libtextclassifier/lang_id/common/math/ |
| D | softmax.cc | 75 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/libtextclassifier/utils/math/ |
| D | softmax.cc | 77 std::vector<float> softmax; in ComputeSoftmax() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | activations.py | 44 def softmax(x, axis=-1): function
|
| D | backend.py | 3854 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/examples/android/src/org/tensorflow/demo/ |
| D | TensorFlowYoloDetector.java | 118 private void softmax(final float[] vals) { in softmax() method in TensorFlowYoloDetector
|
| /external/libtextclassifier/lang_id/ |
| D | lang-id.cc | 142 std::vector<float> softmax = ComputeSoftmax(scores); in FindLanguages() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 75 auto softmax = in Compile() local
|
| /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/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | vector_diffeomixture.py | 1052 def softmax(x, axis, name=None): function
|
| /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 | 1265 auto softmax = ops::Softmax(root, input); in BM_ImageNetSoftmaxFwd() local
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 2887 def softmax(logits, scope=None): function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2875 def softmax(logits, axis=None, name=None, dim=None): function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 2562 Output softmax = ops::Softmax(s.WithOpName("softmax"), x); in TEST_F() local
2594 Output softmax = ops::Softmax(s.WithOpName("softmax"), floats); in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 981 auto* softmax = new SoftmaxOperator; in ConvertSoftmaxOperator() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 749 string softmax; member
|