- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for SparseSoftmaxCrossEntropyWithLogits (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_space_to_depth_pass.mlir
%19 = "tf.Reshape"(%18, %7) : (tensor<2x1001xf32>, tensor<2xi32>) -> tensor<2x1001xf32> %loss, %backprop = "tf.SparseSoftmaxCrossEntropyWithLogits"(%19, %13) : (tensor<2x1001xf32>, tensor<2xi64>) -> (tensor<2xf32>, tensor<2x1001xf32>) %20 = "tf.Sum"(%loss, %6) {keep_dims = false} : (tensor<2xf32>, tensor<1xi32>) -> tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 37.4K 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)