- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SparseSoftmaxCrossEntropyWithLogits (0.73 sec)
-
tensorflow/c/experimental/gradients/nn_grad_test.cc
AbstractTensorHandle* loss; AbstractTensorHandle* backprop; TF_RETURN_IF_ERROR(ops::SparseSoftmaxCrossEntropyWithLogits( ctx, inputs[0], inputs[1], &loss, &backprop, "SparseSoftmaxCrossEntropyWithLogits")); // `gradient_checker` only works with model that returns only 1 tensor. // Although, `ops::SparseSoftmaxCrossEntropyWithLogits` returns 2 tensors, the // second tensor isn't needed for computing gradient so we could safely drop
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 13:53:47 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
// clang-format on {"PWRED", {"ArgMax", "ArgMin", "DiagPart", "Softmax", "SparseSoftmaxCrossEntropyWithLogits", "LogSoftmax"}}, {"REDUCEWINDOW", {"ArgMax", "ArgMin", "DiagPart", "Softmax", "SparseSoftmaxCrossEntropyWithLogits", "LogSoftmax"}}, {"REDUCEWINDOWPW", {"BiasAddGrad", "LRN", "LRNGrad"}}, {"BN",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
FunctionsAcceptingNoneForIndicesMap() { static auto* const m = new std::unordered_map<string, std::unordered_set<int>>({ {"SoftmaxCrossEntropyWithLogits", {1}}, {"SparseSoftmaxCrossEntropyWithLogits", {1}}, {"FusedBatchNorm", {1, 2, 3, 4}}, }); return m; } } // namespace // If over kMinAggregateCount gradients are accumulated and the total
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
TF_DerivedOperandTypeAttr Tidx = TF_DerivedOperandTypeAttr<1>; TF_DerivedOperandTypeAttr Tsegmentids = TF_DerivedOperandTypeAttr<2>; } def TF_SparseSoftmaxCrossEntropyWithLogitsOp : TF_Op<"SparseSoftmaxCrossEntropyWithLogits", [Pure]> { let summary = [{ Computes softmax cross entropy cost and gradients to backpropagate. }]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)