Searched refs:logits_value (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.h | 109 Value result_value, Value logits_value,
115 Value logits_value);
|
| D | legalize_common.cc | 1355 Value result_value, Value logits_value, in convertSoftmaxOp() argument
1371 logits_value.getType().dyn_cast<RankedTensorType>(); in convertSoftmaxOp()
1410 buildRescale(rewriter, op, int32_logits_type, logits_value, 1.0f, in convertSoftmaxOp()
1611 buildRescale(rewriter, op, int32_logits_type, logits_value, 1.0f, in convertSoftmaxOp()
1752 output_type, logits_value); in convertSoftmaxOp()
1774 Value logits_value) { in convertLogSoftmaxOp() argument
1808 output_type, logits_value); in convertLogSoftmaxOp()
|