Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 756 for Reserve (0.2 sec)

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

      llvm::SmallVector<Value, 4> data_fetches;
      llvm::SmallVector<Type, 4> data_types;
      llvm::SmallVector<Value, 4> control_fetches;
      data_fetches.reserve(fetch.getFetches().size());
      data_types.reserve(data_fetches.capacity());
      control_fetches.reserve(data_fetches.capacity());
    
      for (auto value : fetch.getFetches()) {
        if (mlir::isa<ControlType>(value.getType())) {
          control_fetches.push_back(value);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc

      mlir::FunctionType func_type = main_func.getFunctionType();
    
      outputs->clear();
      outputs->reserve(func_type.getNumResults());
      resource_updates->clear();
      resource_updates->reserve(func_type.getNumResults());
    
      std::vector<xla::Shape> shapes;
      shapes.reserve(func_type.getNumResults());
    
      llvm::SmallDenseMap<unsigned, unsigned> output_to_input_alias;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 17:24:39 UTC 2024
    - 45.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc

      std::vector<int32_t> rhs_transpose;
      std::vector<int32_t> out_transpose;
      lhs_transpose.reserve(dnums.lhs_rhs_out.size() + dnums.lhs_out.size() +
                            dnums.lhs_rhs.size());
      rhs_transpose.reserve(dnums.lhs_rhs_out.size() + dnums.rhs_out.size() +
                            dnums.lhs_rhs.size());
      out_transpose.reserve(dnums.lhs_rhs_out.size() + dnums.lhs_out.size() +
                            dnums.rhs_out.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

      llvm::SmallVector<Value, 8> new_packed_inputs;
      llvm::SmallVector<llvm::SmallVector<Value, 8>, 8> replicated_inputs;
      replicated_inputs.reserve(replicate.GetNumReplicatedBlockArguments());
      new_packed_inputs.reserve(replicate.GetNumPackedBlockArguments());
      for (const auto& arg : replicate.GetReplicatedBlockArguments()) {
        replicated_inputs.emplace_back();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  5. pkg/scheduler/framework/interface.go

    type ReservePlugin interface {
    	Plugin
    	// Reserve is called by the scheduling framework when the scheduler cache is
    	// updated. If this method returns a failed Status, the scheduler will call
    	// the Unreserve method for all enabled ReservePlugins.
    	Reserve(ctx context.Context, state *CycleState, p *v1.Pod, nodeName string) *Status
    	// Unreserve is called by the scheduling framework when a reserved pod was
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Fri May 31 15:52:16 UTC 2024
    - 35.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter.cc

      // compilation.
      std::vector<tensorflow::XlaExpression> expressions;
      std::vector<tensorflow::Tensor> tensors;
      std::vector<tensorflow::TensorValue> inputs;
      expressions.reserve(op_->getNumOperands());
      tensors.reserve(op_->getNumOperands());
      inputs.reserve(op_->getNumOperands());
    
      if (failed(
              PrepareKernelInputs(required_consts, expressions, tensors, inputs)))
        return failure();
    
      params_.inputs = inputs;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 18.9K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc

      for (int core = 0; core < num_cores_per_replica; ++core) {
        llvm::SmallVector<StringRef, 8> devices_by_core;
        devices_by_core.reserve(num_replicas);
        llvm::SmallVector<StringRef, 8> hosts_by_core;
        hosts_by_core.reserve(num_replicas);
        for (int replica = 0; replica < num_replicas; ++replica) {
          devices_by_core.push_back(tpu_devices[replica][core].device);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 29.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.cc

            "Length of input node array and data shape doesn't match (#arrays ",
            node_names.size(), ", #input_shapes ", node_shapes.size(), ")"));
      }
    
      // StringMap doesn't support reserve else reserve input map size here.
      for (int i = 0, end = node_names.size(); i < end; i++) {
        auto& name = node_names[i];
        const string& type = used_node_dtypes[i];
        if (name.empty()) continue;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc

      } while (succeeded(parser->parseOptionalComma()));
    
      region_args->reserve(replicated_region_args.size() +
                           packed_region_args.size());
      region_args->append(replicated_region_args.begin(),
                          replicated_region_args.end());
      region_args->append(packed_region_args.begin(), packed_region_args.end());
    
      region_arg_types->reserve(replicated_region_arg_types.size() +
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 33.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/python/saved_model_to_tfl_flatbuffer.cc

      mlir::DictionaryAttr tf_attrs =
          entry_function->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function");
      llvm::SmallVector<llvm::StringRef, 4> function_input_names;
      function_input_names.reserve(model_flags.input_arrays().size());
      auto input_attr = tf_attrs.get("inputs");
      if (!input_attr) {
        return errors::InvalidArgument("no inputs attribute found");
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:39:37 UTC 2024
    - 11K bytes
    - Viewed (0)
Back to top