- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for reduce_max (0.08 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/mnist_train.py
with tf.GradientTape() as tape: logits = model(inputs) loss_value = tf.reduce_mean( tf.nn.softmax_cross_entropy_with_logits(labels, logits)) grads = tape.gradient(loss_value, model.trainable_variables) correct_prediction = tf.equal(tf.argmax(logits, 1), tf.argmax(labels, 1)) accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 20 03:05:18 UTC 2021 - 6.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.h
using OpRewritePattern<TFL::MeanOp>::OpRewritePattern; LogicalResult matchAndRewrite(TFL::MeanOp mean_op, PatternRewriter& rewriter) const override; }; // Insert Requant ops for reduce_mean. struct InsertRequantForReduceMean : public OpRewritePattern<TFL::MeanOp> { using OpRewritePattern<TFL::MeanOp>::OpRewritePattern; LogicalResult matchAndRewrite(TFL::MeanOp mean_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 03 16:37:16 UTC 2022 - 4.3K bytes - Viewed (0)