Searched refs:log_softmax (Results 1 – 25 of 25) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | categorical.py | 319 nn_ops.log_softmax(self.logits) * self.probs, axis=-1)
344 delta_log_probs1 = (nn_ops.log_softmax(a.logits) -
345 nn_ops.log_softmax(b.logits))
|
| D | multinomial.py | 285 return math_ops.reduce_sum(counts * nn_ops.log_softmax(self.logits), -1)
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | log_softmax.py | 42 out = tf.nn.log_softmax(input_tensor)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | softmax_op_test.py | 75 tf_softmax = nn_ops.log_softmax(np_features, axis=dim, name=name)
126 tf_log_softmax = nn_ops.log_softmax(features)
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 96 from tensorflow.lite.testing.op_tests.log_softmax import make_log_softmax_tests
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | categorical_test.py | 344 log_probabilities = nn_ops.log_softmax(logits)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize_patterns.td | 729 (ArgMinMaxOp (TFL_LogSoftmaxOp:$log_softmax $logits),
732 [(HasOneUse $log_softmax),
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 208 name: "log_softmax"
|
| D | tensorflow.math.pbtxt | 256 name: "log_softmax"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3739 def log_softmax(logits, axis=None, name=None, dim=None): function
3764 return _wrap_2d_function(logits, gen_nn_ops.log_softmax, axis, name)
3792 return _wrap_2d_function(logits, gen_nn_ops.log_softmax, axis, name)
|
| D | ctc_ops.py | 656 ilabel_log_probs = nn_ops.log_softmax(logits)
|
| D | nn_grad.py | 540 return grad, _BroadcastMul(grad_loss, -nn_ops.log_softmax(logits))
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 615 return (nn.log_softmax(logits_i), nn.log_softmax(logits_i, axis=0),
616 nn.log_softmax(logits_i, axis=-1))
|
| D | pfor.py | 2043 @RegisterPForWithArgs("LogSoftmax", gen_nn_ops.log_softmax)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 256 name: "log_softmax"
|
| D | tensorflow.nn.pbtxt | 244 name: "log_softmax"
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_NumPy_Text_Generation.ipynb | 804 …"We define the loss function from scratch, using `tf.nn.log_softmax`. (Our definition is the same …
819 " predictions = tf.nn.log_softmax(predictions)\n",
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | BUILD | 518 "reference/integer_ops/log_softmax.h",
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 342 "log_softmax",
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 375 nn_ops.log_softmax,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 245 func @log_softmax(%arg0: tensor<8x16xf32>) -> tensor<8x16xf32> {
248 // CHECK-LABEL: log_softmax
249 // CHECK: "tfl.log_softmax"(%arg0) : (tensor<8x16xf32>) -> tensor<8x16xf32>
|
| D | optimize.mlir | 1625 …%0 = "tfl.log_softmax"(%arg0) {beta = 1.000000e+00 : f32} : (tensor<16x1024xf32>) -> tensor<16x102…
1636 …%0 = "tfl.log_softmax"(%arg0) {beta = 1.000000e+00 : f32} : (tensor<16x1024xf32>) -> tensor<16x102…
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/tests/ |
| D | tfl-to-tosa-pipeline.mlir | 606 %0 = "tfl.log_softmax"(%arg0) : (tensor<13x21x3xf32>) -> tensor<13x21x3xf32>
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 3857 ### tfl.log_softmax
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 1911 def TFL_LogSoftmaxOp : TFL_Op<"log_softmax", [
|