- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for bitcast_op (0.11 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
bcast_dims); } else { return rewriter.notifyMatchFailure(bcast_op, "Unsupported element type."); } Value new_const_op = rewriter.create<mhlo::ConstantOp>(bcast_op.getLoc(), result); rewriter.replaceOp(bcast_op, {new_const_op}); return success(); } }; using FoldBroadcastInDimBeforeMulOp =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
Value add_op_result = op->getResult(0); // Broadcast bias value if unmatched with output shape. auto bcast_op = TryCast<BroadcastInDimOp>(bias_op.getDefiningOp(), /*name=*/"broadcast_in_dim_op"); if (failed(bcast_op)) { bcast_op = TryCast<DynamicBroadcastInDimOp>( bias_op.getDefiningOp(), /*name=*/"dynamic_broadcast_in_dim_op"); }
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/quantization/tensorflow/passes/quantize.cc
auto round_val = rewriter.create<TF::RoundOp>( sc_op.getLoc(), float_avg_pool_op.getOutput()); auto icast_op = rewriter.create<TF::CastOp>( sc_op.getLoc(), q_result_type.clone(qtype.getStorageType()), round_val); avg_pool_op.getResult().replaceAllUsesWith(icast_op.getResult()); return success(); } }; // Applies quantization on the model in TF dialect. class QuantizePass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::AnyOp>(), TypeID::get<TF::AvgPoolOp>(), TypeID::get<TF::AvgPool3DGradOp>(), TypeID::get<TF::AvgPoolGradOp>(), TypeID::get<TF::BatchToSpaceNDOp>(), TypeID::get<TF::BitcastOp>(), TypeID::get<TF::BroadcastToOp>(), TypeID::get<TF::CollectivePermuteOp>(), TypeID::get<TF::ComplexOp>(), TypeID::get<TF::ConcatV2Op>(), TypeID::get<TF::ConjOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0)