- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for setInsertionPointAfterValue (0.21 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
RankedTensorType::get({effective_num_bins}, rewriter.getI64Type()), }; // Insert custom aggregation op between operand and operator. rewriter.setInsertionPointAfterValue(value); Operation *aggregator_op = rewriter.create<TF::CustomAggregatorOp>( op->getLoc(), output_types, value, attributes); Value aggregator_op_result = aggregator_op->getOpResult(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
return success(); } void rewrite(LiftedOpT op, PatternRewriter &rewriter) const override { // Only support ops with 1 results Value result = op->getResult(0); rewriter.setInsertionPointAfterValue(result); // In Whole model, we first need to set file_name as // unquantized_tensor_data.pb as it is used by unquantized dump model.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
llvm::MapVector<Operation*, llvm::SmallVector<unsigned int, 4>> arguments_to_erase; for (auto [var_handle_op, resource_tensor] : llvm::zip(variables, resource_tensors_or.value())) { builder.setInsertionPointAfterValue(var_handle_op); auto elements_attr = GetTensorValueAsElementsAttr( var_handle_op, resource_tensor, mgr, builder); ReplaceVarWithConstant(var_handle_op, elements_attr, &arguments_to_erase); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0)