| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | ctc_ops.py | 78 logits=None): argument
201 logits=None, argument
675 def ctc_loss_and_grad(logits, labels, label_length, logit_length, unique=None): argument
743 def _ctc_loss_op_standard(labels, logits, logit_length, logits_time_major, argument
761 def _ctc_loss_op_cudnn(labels, logits, logit_length, logits_time_major, argument
787 logits, argument
883 logits, argument
990 logits, argument
|
| H A D | nn_ops.py | 3747 def _flatten_outer_dims(logits): argument
3862 def softmax_v2(logits, axis=None, name=None): argument
3904 def softmax(logits, axis=None, name=None, dim=None): argument
3918 def log_softmax(logits, axis=None, name=None, dim=None): argument
3948 def log_softmax_v2(logits, axis=None, name=None): argument
3974 def _ensure_xent_args(name, sentinel, labels, logits): argument
3987 def softmax_cross_entropy_with_logits_v2(labels, logits, axis=-1, name=None): argument
4051 def softmax_cross_entropy_with_logits_v2_helper( argument
4183 def softmax_cross_entropy_with_logits( argument
4249 def _sparse_softmax_cross_entropy_with_rank_2_logits(logits, labels, name): argument
[all …]
|
| H A D | random_ops.py | 475 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): argument
505 def categorical(logits, num_samples, dtype=None, seed=None, name=None): argument
533 def multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
|
| H A D | stateless_random_ops.py | 827 def stateless_multinomial(logits, argument
869 def stateless_categorical(logits, argument
910 def stateless_multinomial_categorical_impl(logits, num_samples, dtype, seed): argument
|
| H A D | nn_impl.py | 112 def sigmoid_cross_entropy_with_logits( # pylint: disable=invalid-name argument
156 def sigmoid_cross_entropy_with_logits_v2( # pylint: disable=invalid-name argument
254 def weighted_cross_entropy_with_logits_v2(labels, logits, pos_weight, argument
347 def weighted_cross_entropy_with_logits(labels=None, argument
|
| H A D | nn_xent_test.py | 36 def _SigmoidCrossEntropyWithLogits(self, logits, targets): argument
112 def _WeightedCrossEntropy(self, logits, targets, pos_coeff): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_xent_op_test_base.py | 36 def _opFwdBwd(self, labels, logits): argument
45 def _npXent(self, labels, logits): argument
211 def xent(logits): argument
243 def xent_grad(logits): argument
267 def _testHighDim(self, labels, logits): argument
|
| H A D | sparse_xent_op_test.py | 39 def _sparse_vs_dense_xent_benchmark_dense(labels, logits): argument
58 def _sparse_vs_dense_xent_benchmark_sparse(labels, logits): argument
|
| /aosp_15_r20/external/pytorch/torch/distributions/ |
| H A D | relaxed_categorical.py | 43 def __init__(self, temperature, probs=None, logits=None, validate_args=None): argument
69 def logits(self): member in ExpRelaxedCategorical
122 def __init__(self, temperature, probs=None, logits=None, validate_args=None): argument
137 def logits(self): member in RelaxedOneHotCategorical
|
| H A D | relaxed_bernoulli.py | 44 def __init__(self, temperature, probs=None, logits=None, validate_args=None): argument
81 def logits(self): member in LogitRelaxedBernoulli
134 def __init__(self, temperature, probs=None, logits=None, validate_args=None): argument
147 def logits(self): member in RelaxedBernoulli
|
| H A D | geometric.py | 47 def __init__(self, probs=None, logits=None, validate_args=None): argument
99 def logits(self): member in Geometric
|
| H A D | multinomial.py | 61 def __init__(self, total_count=1, probs=None, logits=None, validate_args=None): argument
90 def logits(self): member in Multinomial
|
| H A D | negative_binomial.py | 38 def __init__(self, total_count, probs=None, logits=None, validate_args=None): argument
90 def logits(self): member in NegativeBinomial
|
| H A D | categorical.py | 53 def __init__(self, probs=None, logits=None, validate_args=None): argument
97 def logits(self): member in Categorical
|
| H A D | binomial.py | 51 def __init__(self, total_count=1, probs=None, logits=None, validate_args=None): argument
107 def logits(self): member in Binomial
|
| H A D | bernoulli.py | 44 def __init__(self, probs=None, logits=None, validate_args=None): argument
93 def logits(self): member in Bernoulli
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | multinomial.cc | 38 const FloatType* logits, int logits_size, in MultinomialSample()
77 const FloatType* logits, int logits_size, in MultinomialSample()
101 const TfLiteTensor* logits, int logits_offset, in MultinomialSample()
175 const TfLiteTensor* logits = tflite::GetInput(context, node, 0); in Eval() local
|
| H A D | multinomial_test.cc | 57 MultinomialOpModel(tflite::TensorData logits, int num_samples, in MultinomialOpModel()
70 int logits() { return logits_; } in logits() function in tflite::__anon752fffe90111::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
|
| H A D | random_ops.cc | 100 const float* logits, size_t logits_size, in GenerateMultinomialNumbers()
204 const TfLiteTensor* logits; in PrepareMultinomial() local
281 const float* logits = GetTensorData<float>(logits_tensor); in EvalMultinomial() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/losses/ |
| H A D | losses_impl.py | 328 def hinge_loss(labels, logits, weights=1.0, scope=None, argument
773 multi_class_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
839 onehot_labels, logits, weights=1.0, label_smoothing=0, scope=None, argument
1048 labels, logits, weights=1.0, scope=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/ |
| H A D | bernoulli.py | 48 logits=None, argument
100 def logits(self): member in Bernoulli
140 def _broadcast(logits, event): argument
|
| H A D | categorical.py | 159 logits=None, argument
246 def logits(self): member in Categorical
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A 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()
|
| /aosp_15_r20/external/executorch/examples/models/llama/experimental/ |
| H A D | generate.py | 25 def logits_to_probs(logits, temperature: float = 1.0, top_k: Optional[int] = None): argument
36 def sample(logits, temperature: float = 1.0, top_k: Optional[int] = None): argument
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| H A D | GpuSoftmax.cpp | 71 TensorInfo logits(src->clone()->set_tensor_shape(logits_sum_shape)); in is_supported_op() local
139 …ITensorInfo *logits = sketch.implementation().create_auxiliary_tensor(src->clone()->set_… in create_op() local
|