- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for perm_values (0.11 sec)
-
tensorflow/compiler/mlir/lite/utils/utils.h
auto perm_values = perm_values_attr.getValues<APInt>(); size_t idx = 0; for (; idx < perm_values_attr.size() - 2; ++idx) { if (perm_values[idx].getSExtValue() != idx) return false; } return (perm_values[idx].getSExtValue() == perm_values_attr.size() - 1) && (perm_values[idx + 1].getSExtValue() == idx); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
// Transpose and constant-fold the weight if needed. if (adj_y.getValue()) { SmallVector<int32_t> perm_values(num_batch_dim); absl::c_iota(perm_values, 0); perm_values.push_back(num_batch_dim + 1); perm_values.push_back(num_batch_dim); Value perm = Create1DConstValue<int32_t>(builder, loc, perm_values); auto transpose_op = builder.create<TF::TransposeOp>(loc, weight, perm);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0)