- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for broadcast_in_dim_v1 (0.3 sec)
-
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)