- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for broadcast_in_dim_v1 (0.31 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/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
Value subtract_op_second_operand = subtract_op->getOperand(1); if (auto broadcast_in_dim_op = TryCast<stablehlo::BroadcastInDimOp>( subtract_op_second_operand.getDefiningOp(), /*name=*/"broadcast_in_dim_for_subtract_op_operand"); succeeded(broadcast_in_dim_op)) { subtract_op_second_operand = broadcast_in_dim_op->getOperand(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K 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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// If no such constant op exists, returns a nullptr. Operation* GetBiasConstOp(Operation* op) { Operation* bias_const_op; if (Operation* broadcast_in_dim_op = FindOperandOfType<stablehlo::BroadcastInDimOp>(op); broadcast_in_dim_op != nullptr) { bias_const_op = broadcast_in_dim_op->getOperand(0).getDefiningOp(); } else { bias_const_op = FindOperandOfType<stablehlo::ConstantOp>(op); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0)