- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for perm_values (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
SmallVector<int64_t, 8> perm0_values; for (const auto &value : perm0.getValues<APInt>()) perm0_values.push_back(value.getSExtValue()); SmallVector<int64_t, 8> perm1_values; for (const auto &value : perm1.getValues<APInt>()) perm1_values.push_back(value.getSExtValue()); for (int i = 0; i < perm0_values.size(); ++i) { if (perm0_values[perm1_values[i]] != i) return false; } return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.td
$asymmetric_quantize_inputs_bmm ), [(AreLastTwoDimsTransposed $perm_value), (IsNoneType $bias)]>; // Fuses TFL_FullyConnectedOp and TFL_TransposeOp Rhs to TFL_BatchMatMulOp def FuseTransposeFCRhsToBatchMatmul : Pat< (TFL_FullyConnectedOp 2DTensorOf<[F32]>:$lhs, (TFL_TransposeOp TensorOf<[F32]>:$rhs, (Arith_ConstantOp:$perm_value $p0)), $bias, $TFL_AF_None, $TFL_FCWO_Default,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 09 23:44:09 UTC 2023 - 2.6K bytes - Viewed (0)