- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for broadcast_in_dim_v1 (0.25 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
%0 = "vhlo.broadcast_in_dim_v1"(%arg0) <{broadcast_dimensions = #vhlo.tensor_v1<dense<[0, 1, 2]> : tensor<3xi64>>}> : (tensor<1x32x256xf32>) -> tensor<4x32x256xf32> return %0 : tensor<4x32x256xf32> } //CHECK:func.func private @broadcast_in_dim(%arg0: tensor<1x32x256xf32>) -> tensor<4x32x256xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
// `BroadcastInDimOp`'s operand is a block argument. absl::Nullable<Operation*> SkipUpwardsOptionalBroadcastInDimOp( absl::Nonnull<Operation*> op) { if (auto broadcast_in_dim_op = dyn_cast_or_null<BroadcastInDimOp>(op); broadcast_in_dim_op != nullptr) { return broadcast_in_dim_op.getOperand().getDefiningOp(); } return op; } class DeferActivationTransposeForAddOp : public OpRewritePattern<AddOp> { public:
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/quantization/stablehlo/passes/quantization_patterns.cc
// ``` template <typename T> Operation* GetBroadcastedUserOp(Operation* op) { // Broadcast bias for known input shape. auto broadcast_in_dim_op = FindUserOfType<BroadcastInDimOp>(op); if (broadcast_in_dim_op != nullptr) { auto target_op = FindUserOfType<T>(broadcast_in_dim_op); if (target_op != nullptr) return target_op; } // Broadcast bias for unknown input shape.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0)