- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for DenseI64ArrayAttr (0.22 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
[&](std::optional<ArrayRef<int64_t>> vals) -> DenseI64ArrayAttr { return vals.has_value() ? DenseI64ArrayAttr::get( ctx, InvertPermutationToVector(*vals, transpose_perm)) : nullptr; }; DenseI64ArrayAttr new_window_dimensions = getInvertPermutedAttr(reduce_op.getWindowDimensions()); DenseI64ArrayAttr new_window_strides =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
// `array_attr` and `permutation` must be equal. Returns a null attribute // if `array_attr` is null. DenseI64ArrayAttr PermuteI64ArrayAttr( PatternRewriter& rewriter, const std::optional<ArrayRef<int64_t>> array_attr, const ArrayRef<int64_t> permutation) const { if (!array_attr.has_value()) return DenseI64ArrayAttr(nullptr); return rewriter.getDenseI64ArrayAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
op.getLoc(), op.getMean().getType(), alpha, op.getMean(), /*broadcast_dimensions=*/DenseI64ArrayAttr()); auto beta_mul_batch_mean = rewriter.create<chlo::BroadcastMulOp>( op.getLoc(), batch_mean.getType(), beta, batch_mean, /*broadcast_dimensions=*/DenseI64ArrayAttr()); batch_mean = rewriter.create<chlo::BroadcastAddOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (1)