| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| 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
|
| 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
|
| /external/tensorflow/tensorflow/python/ops/ |
| 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
|
| 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
|
| D | nn_xent_test.py | 36 def _SigmoidCrossEntropyWithLogits(self, logits, targets): argument
112 def _WeightedCrossEntropy(self, logits, targets, pos_coeff): argument
|
| 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
|
| 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 …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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
|
| D | multinomial_test.cc | 57 MultinomialOpModel(tflite::TensorData logits, int num_samples, in MultinomialOpModel()
70 int logits() { return logits_; } in logits() function in tflite::__anona99f41580111::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
|
| 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
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | bernoulli.py | 48 logits=None, argument
100 def logits(self): member in Bernoulli
140 def _broadcast(logits, event): argument
|
| D | categorical.py | 159 logits=None, argument
246 def logits(self): member in Categorical
|
| D | multinomial.py | 158 logits=None, argument
221 def logits(self): member in Multinomial
|
| /external/tensorflow/tensorflow/python/ops/losses/ |
| 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
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| 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
|
| /external/tensorflow/tensorflow/python/framework/experimental/ |
| D | nn_ops.py | 26 def sparse_softmax_cross_entropy_with_logits(logits, labels, name=None): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | multinomial_op_test.py | 36 def composed_sampler(logits, num_samples): argument
146 def _do_sampling(self, logits, num_samples, sampler): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 47 xla::XlaOp logits, xla::XlaOp labels) { in CrossEntropyWithLogits()
108 auto logits = ctx->Input(0); in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | xent_op_test_base.py | 41 def _opFwdBwd(self, labels, logits, axis=-1): argument
50 def _npXent(self, labels, logits, dim=-1): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | xent_op_test_base.py | 37 def _opFwdBwd(self, labels, logits, axis=-1): argument
46 def _npXent(self, labels, logits, dim=-1): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | xent_op_test.cc | 27 Tensor logits(type, TensorShape({batch_size, num_classes})); in Xent() local
|
| D | sparse_xent_op_test.cc | 29 Tensor logits(type, TensorShape({batch_size, num_classes})); in SparseXent() local
|
| D | sparse_xent_op.cc | 60 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/tpu/ |
| D | async_checkpoint_test.py | 79 def metric_fn(labels, logits): argument
|