| /external/tensorflow/tensorflow/lite/kernels/ |
| D | multinomial.cc | 37 const FloatType* logits, int logits_size, in MultinomialSample()
76 const FloatType* logits, int logits_size, in MultinomialSample()
100 const TfLiteTensor* logits, int logits_offset, in MultinomialSample()
174 const TfLiteTensor* logits = tflite::GetInput(context, node, 0); in Eval() local
|
| D | multinomial_test.cc | 57 MultinomialOpModel(tflite::TensorData logits, int num_samples, in MultinomialOpModel()
70 int logits() { return logits_; } in logits() function in tflite::__anon5b866f7a0111::MultinomialOpModel
188 std::vector<Float> logits(kNumLogits, static_cast<Float>(0.0f)); in TYPED_TEST() local
214 std::vector<Float> logits( in TYPED_TEST() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | ctc_ops.py | 82 logits=None): argument
205 logits=None, argument
629 def ctc_loss_and_grad(logits, labels, label_length, logit_length, unique=None): argument
697 def _ctc_loss_op_standard(labels, logits, logit_length, logits_time_major, argument
715 def _ctc_loss_op_cudnn(labels, logits, logit_length, logits_time_major, argument
741 logits, argument
836 logits, argument
940 logits, argument
|
| D | stateless_random_ops.py | 584 def stateless_multinomial(logits, argument
625 def stateless_categorical(logits, argument
665 def stateless_multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
|
| D | random_ops.py | 472 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
501 def categorical(logits, num_samples, dtype=None, seed=None, name=None): argument
528 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 | 3550 def _flatten_outer_dims(logits): argument
3666 def softmax(logits, axis=None, name=None, dim=None): argument
3710 def softmax_v2(logits, axis=None, name=None): argument
3739 def log_softmax(logits, axis=None, name=None, dim=None): argument
3769 def log_softmax_v2(logits, axis=None, name=None): argument
3795 def _ensure_xent_args(name, sentinel, labels, logits): argument
3806 def softmax_cross_entropy_with_logits_v2(labels, logits, axis=-1, name=None): argument
3870 def softmax_cross_entropy_with_logits_v2_helper( argument
3998 def softmax_cross_entropy_with_logits( argument
4066 def sparse_softmax_cross_entropy_with_logits( argument
[all …]
|
| D | nn_impl.py | 115 def sigmoid_cross_entropy_with_logits( # pylint: disable=invalid-name argument
198 def sigmoid_cross_entropy_with_logits_v2( # pylint: disable=invalid-name argument
249 def weighted_cross_entropy_with_logits_v2(labels, logits, pos_weight, argument
329 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/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 43 xla::XlaOp logits, bool log) { in BuildSoftmaxCustomCall()
69 auto logits = ctx->Input(0); in Compile() local
118 xla::XlaOp logits, xla::XlaOp labels) { in CrossEntropyWithLogits()
179 auto logits = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| D | losses_impl.py | 321 def hinge_loss(labels, logits, weights=1.0, scope=None, argument
659 multi_class_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
725 onehot_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
848 labels, logits, weights=1.0, scope=None, argument
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | nn_ops.py | 30 def sparse_softmax_cross_entropy_with_logits(logits, labels, name=None): argument
|
| /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/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
|
| D | softmax_op_gpu.cu.cc | 72 __global__ void GenerateNormalizedProb(const T* logits, const U* sum_probs, in GenerateNormalizedProb()
111 const Eigen::half* logits, const float* sum_probs, in GenerateNormalizedProb()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_xent_op_test.py | 305 def _sparse_vs_dense_xent_benchmark_dense(labels, logits): argument
324 def _sparse_vs_dense_xent_benchmark_sparse(labels, logits): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint_test.py | 83 def metric_fn(labels, logits): argument
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | gradient_checkpoint_test.py | 71 def _compute_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/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 347 const T* logits, in CalculateRowSoftmax()
372 const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel2D()
397 const int* row_ptr, const T* logits, T* softmax) { in CSRSparseMatrixSoftmaxKernel3D()
420 const CSRSparseMatrix& logits, in CSRSparseMatrixSoftmaxGPUImpl()
|
| /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
|