Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 3,700 for Reserve (0.2 sec)

  1. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

    std::optional<ConstBytesAttr> CustomOptions(MLIRContext* context,
                                                mlir::Operation* op) {
      if (auto reserve =
              llvm::dyn_cast_or_null<mlir::TF::TensorListReserveOp>(op)) {
        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(reserve.getElementDtype());
    
        return CreateListReserveOptions(context, tflite_type);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

      llvm::SmallVector<mlir::APInt> ret;
      size_t read_size = sizeof(T);
      int bytes_len = bytes.size();
      assert(bytes_len % read_size == 0);
    
      int elem_count = bytes_len / read_size;
      ret.reserve(elem_count);
    
      const char* data_ptr = reinterpret_cast<const char*>(bytes.data());
      for (int i = 0; i < elem_count; i++) {
        T val = llvm::support::endian::readNext<T, llvm::endianness::native,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_launch_util_test.cc

                                                  device_->parsed_name().id));
    
        std::vector<xla::PjRtBuffer*> executable_args;
        executable_args.reserve(result->input_mapping.size());
        absl::flat_hash_set<int> non_donatable_input_indices;
        TF_EXPECT_OK(PreparePjRtExecutableArguments(
            /*num_missing_prefix_ctx_inputs=*/0, result->input_mapping, inputs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 28.8K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc

        const mlir::Value& original_source, mlir::OpBuilder* builder,
        llvm::SmallVectorImpl<mlir::Value>* tiled_inputs) {
      llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input;
      split_ops_for_tiled_input.reserve(
          input_sharding.tile_assignment_devices_size());
    
      // Creates a tree of split nodes for sharding tiled inputs. Splits nodes
      // are created such that input data is sharded in row major order.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 22 21:28:13 UTC 2024
    - 34K bytes
    - Viewed (0)
  5. pkg/scheduler/testing/framework/fake_plugins.go

    		}, nil
    	}
    }
    
    // FakeReservePlugin is a test reserve plugin.
    type FakeReservePlugin struct {
    	Status *framework.Status
    }
    
    // Name returns name of the plugin.
    func (pl *FakeReservePlugin) Name() string {
    	return "FakeReserve"
    }
    
    // Reserve invoked at the Reserve extension point.
    func (pl *FakeReservePlugin) Reserve(_ context.Context, _ *framework.CycleState, _ *v1.Pod, _ string) *framework.Status {
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Mon Feb 26 19:07:19 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc

        }
        return FoldAdaptor(operation);
      }
    
      // Gets a list of ElementsAttr behind each constant operand.
      llvm::SmallVector<ElementsAttr> OperandData() {
        llvm::SmallVector<ElementsAttr> res;
        res.reserve(operation_->getNumOperands());
        for (auto opr : operation_->getOperands()) {
          auto op = llvm::dyn_cast<stablehlo::ConstantOp>(opr.getDefiningOp());
          res.push_back(op.getValue());
        }
        return res;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 06:11:55 UTC 2024
    - 4.5K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

                                                     ShapedType type) {
      // Extract to a vector of StringRefs for converting.
      auto arr = input_tensor.flat<tstring>();
      std::vector<mlir::StringRef> string_refs;
      string_refs.reserve(arr.size());
      for (int i = 0; i < arr.size(); i++) {
        const auto& val = arr(i);
        string_refs.push_back({val.data(), val.size()});
      }
    
      return ElementsAttr(DenseStringElementsAttr::get(type, string_refs));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
  8. tensorflow/c/eager/unified_api_testutil.cc

        absl::flat_hash_set<int> null_indices;
        {
          AbstractContextPtr func_ctx(BuildFunction(fn_name));
          std::vector<AbstractTensorHandle*> func_inputs;
          func_inputs.reserve(inputs.size());
          TF_RETURN_IF_ERROR(
              CreateParamsForInputs(func_ctx.get(), inputs, &func_inputs));
          std::vector<AbstractTensorHandle*> model_outputs;
          model_outputs.resize(outputs.size());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Feb 27 13:57:45 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  9. pkg/scheduler/metrics/metrics.go

    	ScoreExtensionNormalize     = "ScoreExtensionNormalize"
    	PreBind                     = "PreBind"
    	Bind                        = "Bind"
    	PostBind                    = "PostBind"
    	Reserve                     = "Reserve"
    	Unreserve                   = "Unreserve"
    	Permit                      = "Permit"
    )
    
    // All the histogram based metrics have 1ms as size for the smallest bucket.
    var (
    Registered: Sat Jun 15 01:39:40 UTC 2024
    - Last Modified: Sat Apr 27 08:22:53 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc

      quant::QuantizedType convertFakeQuantAttrsToType(ConstFakeQuantPerAxis fqOp,
                                                       Type expressedType) const {
        SmallVector<double, 4> min, max;
        min.reserve(fqOp.getMin().size());
        max.reserve(fqOp.getMax().size());
        for (auto m : fqOp.getMin())
          min.push_back(cast<FloatAttr>(m).getValueAsDouble());
        for (auto m : fqOp.getMax())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 02:10:16 UTC 2024
    - 6K bytes
    - Viewed (0)
Back to top