- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for setInsertionPointAfterValue (0.32 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/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)