Search Options

Results per page
Sort
Preferred Languages
Advance

Results 111 - 120 of 241 for set_type (0.18 sec)

  1. tensorflow/compiler/mlir/tfrt/analysis/cost_analysis.cc

      auto value_type = mlir::cast<mlir::TensorType>(op.getValues().getType());
      auto key_type = mlir::cast<mlir::TensorType>(op.getKeys().getType());
    
      int64_t output_size = InferTensorSize(context, value_type);
    
      int64_t cost = kLookupTableFindCostScale * output_size;
    
      if (mlir::isa<mlir::TF::StringType>(key_type.getElementType()))
        cost *= kLookupTableFindStringKeyCostScale;
    
      return cost;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      }
      if (refined_shape || lhs_element_type != lhs_shape_type.getElementType()) {
        Type new_type;
        if (!lhs_shape_type.hasRank() && !rhs_shape_type.hasRank())
          new_type = UnrankedTensorType::get(lhs_element_type);
        else
          new_type = lhs_shape_type.clone(shape, lhs_element_type);
        DCOMMENT("Refined to: " << new_type);
        return new_type;
      }
      DCOMMENT("No refinement " << lhs);
      return lhs;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/ops/array_ops.h

                     const char* raw_device_name = nullptr);
    
    // Returns the shape of a tensor.
    Status Shape(AbstractContext* ctx, AbstractTensorHandle* const input,
                 AbstractTensorHandle** output, DataType out_type = DT_INT32,
                 const char* name = nullptr, const char* raw_device_name = nullptr);
    
    // Inserts a dimension of 1 into a tensor's shape.
    Status ExpandDims(AbstractContext* ctx, AbstractTensorHandle* const input,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 10 19:11:36 UTC 2022
    - 2.7K bytes
    - Viewed (0)
  4. src/cmd/compile/internal/ir/func.go

    		base.ErrorfAt(pos, 0, "internal/abi.FuncPC%s does not accept func expression, which is ABIInternal", wantABI)
    	}
    	var e Node = NewUnaryExpr(pos, OIDATA, n)
    	e.SetType(types.Types[types.TUINTPTR].PtrTo())
    	e.SetTypecheck(1)
    	e = NewStarExpr(pos, e)
    	e.SetType(types.Types[types.TUINTPTR])
    	e.SetTypecheck(1)
    	return e
    }
    
    // DeclareParams creates Names for all of the parameters in fn's
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu May 23 01:05:44 UTC 2024
    - 21.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

            addn.getSum().setType(
                mlir::cast<TensorType>(addn.getOperand(0).getType()));
            auto size = it->getSecond();
            (*buffer_to_size)[addn.getSum()] = size;
          }
        } else if (auto zeros = llvm::dyn_cast<TF::ZerosLikeOp>(&op)) {
          if (buffer_to_size->count(zeros.getX()) > 0) {
            zeros.getY().setType(zeros.getX().getType());
            auto size = (*buffer_to_size)[zeros.getX()];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  6. src/cmd/link/internal/ld/macho.go

    		sb.SetType(sym.SMACHOPLT)
    		sb.SetReachable(true)
    
    		s = ctxt.loader.LookupOrCreateSym(".got", 0) // will be __nl_symbol_ptr
    		sb = ctxt.loader.MakeSymbolUpdater(s)
    		sb.SetType(sym.SMACHOGOT)
    		sb.SetReachable(true)
    		sb.SetAlign(4)
    
    		s = ctxt.loader.LookupOrCreateSym(".linkedit.plt", 0) // indirect table for .plt
    		sb = ctxt.loader.MakeSymbolUpdater(s)
    		sb.SetType(sym.SMACHOINDIRECTPLT)
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Mon May 20 15:32:53 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc

          xla::ClientLibrary::GetOrCreateCompileOnlyClient(platform).value();
    
      std::vector<TensorShape> arg_shapes = {{1}};
      TPUCompileMetadataProto metadata_proto;
      auto arg = metadata_proto.add_args();
      arg->set_dtype(DataType::DT_FLOAT);
      arg->set_kind(TPUCompileMetadataProto::Arg::PARAMETER);
      metadata_proto.add_retvals();
      bool use_tuple_args = true;
      std::vector<ShardingAndIndex> arg_core_mapping;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Apr 14 20:29:34 UTC 2024
    - 6K bytes
    - Viewed (0)
  8. maven-plugin-api/src/main/java/org/apache/maven/plugin/descriptor/PluginDescriptorBuilder.java

            ComponentDependency cd = new ComponentDependency();
    
            cd.setArtifactId(extractArtifactId(d));
    
            cd.setGroupId(extractGroupId(d));
    
            cd.setType(d.getChild("type").getValue());
    
            cd.setVersion(extractVersion(d));
            return cd;
        }
    
        @SuppressWarnings("checkstyle:methodlength")
    Registered: Wed Jun 12 09:55:16 UTC 2024
    - Last Modified: Tue Jun 11 07:23:04 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  9. src/cmd/link/internal/riscv64/asm.go

    			// loader symbols, in order to reduce memory consumption. This
    			// would require changes to genelfsym so that it called
    			// putelfsym and putelfsyment as appropriate.
    			sb := ldr.MakeSymbolBuilder(fakeLabelName)
    			sb.SetType(sym.STEXT)
    			sb.SetValue(ldr.SymValue(s) + int64(r.Off()))
    			sb.SetLocal(true)
    			sb.SetReachable(true)
    			sb.SetVisibilityHidden(true)
    			sb.SetSect(ldr.SymSect(s))
    			if outer := ldr.OuterSym(s); outer != 0 {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri Mar 01 08:06:08 UTC 2024
    - 22.8K bytes
    - Viewed (0)
  10. src/cmd/link/internal/ld/data.go

    	types.SetType(sym.STYPE)
    	types.SetSize(8)
    	ldr.SetAttrSpecial(types.Sym(), false)
    
    	etypes := ldr.CreateSymForUpdate("runtime.etypes", 0)
    	etypes.SetType(sym.SFUNCTAB)
    	ldr.SetAttrSpecial(etypes.Sym(), false)
    
    	if ctxt.HeadType == objabi.Haix {
    		rodata := ldr.CreateSymForUpdate("runtime.rodata", 0)
    		rodata.SetType(sym.SSTRING)
    		rodata.SetSize(8)
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed Jun 12 15:10:50 UTC 2024
    - 100.5K bytes
    - Viewed (0)
Back to top