- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 59 for setLoc (0.24 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/fold_constants_to_subgraph.cc
(llvm::isa<TFL::ConstOp, TFL::QConstOp, arith::ConstantOp>(const_op)) && "Expect QConst or Const op."); OpBuilder builder(func.getBody()); auto cloned_const_op = const_op->clone(); cloned_const_op->setLoc(func.getBody().getLoc()); builder.insert(cloned_const_op); // Rewire the usage. func.getArgument(argument_index) .replaceAllUsesWith(cloned_const_op->getResult(0)); } bool IsConstOrQConstInt(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
op->getParentOfType<func::FuncOp>().getSymNameAttr().str(); quantization_unit->set_func_name(func_name); } QuantizationUnitLoc unit_loc(getContext(), quantization_unit.value()); op->setLoc(unit_loc); return success(); } }; void AddQuantizationUnitLocPass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
auto dcast = b.create<quantfork::DequantizeCastOp>( dq.getLoc(), dq.getOutput().getType(), dq.getInput()); dq.getOutput().replaceAllUsesWith(dcast); dq.erase(); } else if (auto q = llvm::dyn_cast<QuantizeOp>(op)) { auto qcast = b.create<quantfork::QuantizeCastOp>( q.getLoc(), q.getOutput().getType(), q.getInput()); q.getOutput().replaceAllUsesWith(qcast);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
auto new_read_variable_op = builder.create<ReadVariableOp>(read_variable_op.getLoc(), ref_qtype, read_variable_op.getResourceId()); auto new_dq_op = builder.create<DequantizeOp>( read_variable_op.getLoc(), read_variable_op.getResult().getType(), new_read_variable_op.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
req_op.getLoc(), func_op.getResultTypes()[0], req_op.getOperand()); if (func_name.contains("_relu6_")) { auto min = rewriter.create<mlir::stablehlo::ConstantOp>( req_op.getLoc(), rewriter.getF32FloatAttr(0)); auto max = rewriter.create<mlir::stablehlo::ConstantOp>( req_op.getLoc(), rewriter.getF32FloatAttr(6));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
auto erf_op = rewriter.create<chlo::ErfOp>(mul_op.getLoc(), mul_op); auto add_op = rewriter.create<stablehlo::AddOp>(erf_op.getLoc(), erf_op, one_cst); auto lhs_mul_op = rewriter.create<stablehlo::MulOp>( half_cst.getLoc(), new_func.getArguments()[0], half_cst); auto output_mul_op = rewriter.create<stablehlo::MulOp>(lhs_mul_op.getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
auto tf_reduce_op = rewriter.create<BooleanReduce>( reduce_op.getLoc(), reduce_op->getResult(0).getType(), operand, reduction_indices, /*keep_dim=*/rewriter.getBoolAttr(false)); auto tf_argreduce_op = rewriter.create<ArgReduce>( reduce_op.getLoc(), reduce_op->getResult(1).getType(), operand, reduction_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0), const_op.getResult()}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
Type output_type = bmm_op.getResult().getType(); auto no_input = rewriter.create<TFL::NoValueOp>( bmm_op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr()); auto fc_op = rewriter.create<TFL::FullyConnectedOp>( bmm_op->getLoc(), ArrayRef<Type>{output_type}, /*input=*/output_lhs, /*filter=*/output_rhs, /*bias=*/no_input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. { Block &entry_block = outlined_func.getBody().front(); auto loc = outlined_func.getLoc(); for (Value operand : operands) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)