Search Options

Results per page
Sort
Preferred Languages
Advance

Results 31 - 40 of 161 for setLoc (0.29 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc

      SendTPUEmbeddingGradientsOp send_op;
      if (failed(GetOp(region, &send_op))) return failure();
    
      // No TPU embedding ops.
      if (!recv_op && !send_op) return success();
    
      Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc();
      StringRef config = recv_op ? recv_op.getConfig() : send_op.getConfig();
    
      // Create XlaRecvTPUEmbeddingDeduplicationData op.
      OpBuilder builder(region);
      auto output_ty =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 22:55:42 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc

      void runOnOperation() override;
    };
    
    void ReplaceClusterReturnWithReturn(tf_device::ReturnOp cluster_return_op,
                                        OpBuilder* builder) {
      builder->create<func::ReturnOp>(cluster_return_op.getLoc(),
                                      cluster_return_op.getOperands());
      cluster_return_op.erase();
    }
    
    // Builds a function that outlines region attached to cluster_op or launch_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc

        } else {
          initial_value = mlir::Attribute();
        }
        opToName[globalTensor] = name;
        auto variableOp = globalBuilder.create<ml_program::GlobalOp>(
            globalTensor.getLoc(), name, globalTensor.getType(),
            globalTensor.getIsMutable(), initial_value,
            /*visibility=*/globalBuilder.getStringAttr("private"));
        variableOp.setPrivate();
      }
    
      SymbolTable syms(module);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.6K bytes
    - Viewed (0)
  4. src/runtime/signal_arm64.go

    	print("fault   ", hex(c.fault()), "\n")
    }
    
    //go:nosplit
    //go:nowritebarrierrec
    func (c *sigctxt) sigpc() uintptr { return uintptr(c.pc()) }
    
    func (c *sigctxt) setsigpc(x uint64) { c.set_pc(x) }
    func (c *sigctxt) sigsp() uintptr    { return uintptr(c.sp()) }
    func (c *sigctxt) siglr() uintptr    { return uintptr(c.lr()) }
    
    // preparePanic sets up the stack to look like a call to sigpanic.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed Apr 05 18:16:00 UTC 2023
    - 3.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc

      // Every ModuleOp has at least one region and one block.
      Block* first_block = &module->getRegion(0).front();
      builder.setInsertionPointToEnd(first_block);
      auto func = builder.create<mlir::func::FuncOp>(
          ops[0]->getLoc(), dialect.str() + std::to_string(function_id),
          builder.getFunctionType(input_types, output_types));
      func->setAttr("dialect", builder.getStringAttr(dialect));
      auto block = func.addEntryBlock();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 8K bytes
    - Viewed (0)
  6. src/cmd/compile/internal/walk/closure.go

    	clos := ir.NewCompLitExpr(base.Pos, ir.OCOMPLIT, typ, nil)
    	clos.SetEsc(clo.Esc())
    	clos.List = append([]ir.Node{ir.NewUnaryExpr(base.Pos, ir.OCFUNC, clofn.Nname)}, closureArgs(clo)...)
    	for i, value := range clos.List {
    		clos.List[i] = ir.NewStructKeyExpr(base.Pos, typ.Field(i), value)
    	}
    
    	addr := typecheck.NodAddr(clos)
    	addr.SetEsc(clo.Esc())
    
    	// Force type conversion from *struct to the func type.
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon Nov 20 15:56:08 UTC 2023
    - 6.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc

        // TODO: Map to a qbarrier with an attribute like [Forced] to signal that
        // this is a forced/hard-coded constraint.
        auto qbarrier = rewriter.create<QuantizeCastOp>(op.getLoc(), quantizedType,
                                                        op.getInputs());
        rewriter.replaceOpWithNewOp<DequantizeCastOp>(op, converter.input_type,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/translate/split_into_island_per_op_pass.cc

      tf_executor::YieldOp yield = island.GetYield();
      if (yield.getNumOperands() == 0) {
        builder.create<TF::NoOp>(island.getLoc(), TypeRange{}, ValueRange{});
      } else if (yield.getNumOperands() == 1) {
        Value operand = yield.getOperand(0);
        auto identity = builder.create<TF::IdentityOp>(island.getLoc(),
                                                       operand.getType(), operand);
        yield.setOperand(0, identity.getOutput());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 7.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc

        auto pi = builder.create<TF::TPUPartitionedInputV2Op>(
            op.getLoc(), op.getType(), op.getOperands(),
            builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false),
            op.get_XlaShardingAttr());
        op->replaceAllUsesWith(pi);
      } else {
        auto po = builder.create<TF::TPUPartitionedOutputV2Op>(
            op.getLoc(), op.getResultTypes(), op.getOperand(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  10. src/runtime/signal_linux_s390x.go

    func (c *sigctxt) set_r13(x uint64)     { c.regs().gregs[13] = x }
    func (c *sigctxt) set_link(x uint64)    { c.regs().gregs[14] = x }
    func (c *sigctxt) set_sp(x uint64)      { c.regs().gregs[15] = x }
    func (c *sigctxt) set_pc(x uint64)      { c.regs().psw_addr = x }
    func (c *sigctxt) set_sigcode(x uint32) { c.info.si_code = int32(x) }
    func (c *sigctxt) set_sigaddr(x uint64) {
    	*(*uintptr)(add(unsafe.Pointer(c.info), 2*goarch.PtrSize)) = uintptr(x)
    }
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Jun 17 20:42:23 UTC 2021
    - 4.5K bytes
    - Viewed (0)
Back to top