| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| D | mnist.py | 86 def loss(logits, labels): argument
130 def evaluation(logits, labels): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | stateless_random_ops.py | 192 def stateless_multinomial(logits, argument
231 def stateless_categorical(logits, argument
269 def stateless_multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
|
| D | ctc_ops.py | 80 logits=None): argument
203 logits=None, argument
615 def ctc_loss_and_grad(logits, labels, label_length, logit_length, unique=None): argument
672 def _ctc_loss_op_standard(labels, logits, logit_length, logits_time_major, argument
690 def _ctc_loss_op_cudnn(labels, logits, logit_length, logits_time_major, argument
715 logits, argument
809 logits, argument
913 logits, argument
|
| D | random_ops.py | 392 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
420 def categorical(logits, num_samples, dtype=None, seed=None, name=None): argument
447 def multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
|
| D | nn_xent_test.py | 40 def _SigmoidCrossEntropyWithLogits(self, logits, targets): argument
116 def _WeightedCrossEntropy(self, logits, targets, pos_coeff): argument
|
| D | nn_ops.py | 2919 def _flatten_outer_dims(logits): argument
2946 def _softmax(logits, compute_op, dim=-1, name=None): argument
2968 def _swap_axis(logits, dim_index, last_index, name=None): argument
3027 def softmax(logits, axis=None, name=None, dim=None): argument
3069 def softmax_v2(logits, axis=None, name=None): argument
3097 def log_softmax(logits, axis=None, name=None, dim=None): argument
3126 def log_softmax_v2(logits, axis=None, name=None): argument
3152 def _ensure_xent_args(name, sentinel, labels, logits): argument
3162 def softmax_cross_entropy_with_logits_v2(labels, logits, axis=-1, name=None): argument
3224 def softmax_cross_entropy_with_logits_v2_helper( argument
[all …]
|
| D | nn_impl.py | 112 def sigmoid_cross_entropy_with_logits( # pylint: disable=invalid-name argument
194 def sigmoid_cross_entropy_with_logits_v2( # pylint: disable=invalid-name argument
244 def weighted_cross_entropy_with_logits_v2(labels, logits, pos_weight, argument
323 def weighted_cross_entropy_with_logits(labels=None, argument
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | bernoulli.py | 52 logits=None, argument
104 def logits(self): member in Bernoulli
144 def _broadcast(logits, event): argument
|
| D | categorical.py | 163 logits=None, argument
250 def logits(self): member in Categorical
|
| D | multinomial.py | 162 logits=None, argument
225 def logits(self): member in Multinomial
|
| /external/tensorflow/tensorflow/examples/tf2_showcase/ |
| D | mnist.py | 125 def loss(logits, labels): argument
131 def compute_accuracy(logits, labels): argument
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 316 def hinge_loss(labels, logits, weights=1.0, scope=None, argument
649 multi_class_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
713 onehot_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
835 labels, logits, weights=1.0, scope=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 55 auto logits = ctx->Input(0); in Compile() local
95 xla::XlaOp logits, xla::XlaOp labels) { in CrossEntropyWithLogits()
152 auto logits = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op_test.cc | 26 Tensor logits(DT_FLOAT, TensorShape({batch_size, num_classes})); in Xent() local
|
| D | sparse_xent_op_test.cc | 28 Tensor logits(DT_FLOAT, TensorShape({batch_size, num_classes})); in SparseXent() local
|
| D | sparse_xent_op.cc | 57 const Tensor& logits = context->input(0); in Compute() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | multinomial_op_test.py | 40 def composed_sampler(logits, num_samples): argument
150 def _do_sampling(self, logits, num_samples, sampler): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_xent_op_test.py | 266 def _sparse_vs_dense_xent_benchmark_dense(labels, logits): argument
285 def _sparse_vs_dense_xent_benchmark_sparse(labels, logits): argument
|
| D | ctc_loss_op_test.py | 881 def ctc_loss_cpu(labels, logits, label_length, logit_length): argument
895 def ctc_loss_gpu(labels, logits, label_length, logit_length): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint_test.py | 83 def metric_fn(labels, logits): argument
|
| /external/tensorflow/tensorflow/python/debug/examples/v2/ |
| D | debug_mnist_v2.py | 202 def loss(logits, labels): argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | categorical_op_test.py | 50 def _do_sampling(self, logits, num_samples): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | interleave_ops.py | 212 def select_dataset_varying_logits(logits, seed): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 80 auto logits = ops::Add(s.WithOpName("logits"), matmul, b2); in CreateMiniGraph() local
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 360 const T* logits, in CalculateRowSoftmax()
385 const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel2D()
410 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D()
433 const CSRSparseMatrix& logits, in CSRSparseMatrixSoftmaxGPUImpl()
|