Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 562 for reserve (0.08 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      std::vector<std::unique_ptr<ParallelTensor>> per_device_outputs;
      per_device_outputs.reserve(first_op_output_count);
      for (int i = 0; i < first_op_output_count; ++i) {
        std::vector<TensorHandlePtr> components;
        components.reserve(underlying_devices_.size());
        for (int j = 0; j < underlying_devices_.size(); ++j) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 25.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device.cc

        return result;
      }
      std::vector<ParallelTensor*> parallel_inputs;
      std::vector<std::unique_ptr<ParallelTensor>> implicitly_broadcast_tensors;
      parallel_inputs.reserve(inputs.size());
      implicitly_broadcast_tensors.reserve(inputs.size());  // not tight
      for (const auto& input : inputs) {
        if (absl::holds_alternative<TFE_TensorHandle*>(input)) {
          if (operation_name == std::string("_EagerConst")) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_function.cc

      if (output_names) {
        output_names_vec.reserve(noutputs);
        for (int i = 0; i < noutputs; ++i) {
          output_names_vec.push_back(string(output_names[i]));
        }
      }
    
      // Process control output names.
      std::vector<string> control_output_names_vec;
      if (control_output_names) {
        control_output_names_vec.reserve(ncontrol_outputs);
        for (int i = 0; i < ncontrol_outputs; ++i) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 13.7K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs;
      std::vector<ParallelTensor*> handle_inputs;
      handle_inputs.reserve(handles.size());
      for (auto& handle : handles) {
        handle_inputs.push_back(handle.get());
      }
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> read_op(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  5. tensorflow/c/eager/abstract_operation.h

    inline absl::Status AbstractOperation::SetAttrStringList(
        const char* attr_name, absl::Span<string const> values) {
      std::vector<const char*> raw_strs;
      std::vector<size_t> lengths;
      raw_strs.reserve(values.size());
      lengths.reserve(values.size());
      for (const auto& s : values) {
        raw_strs.emplace_back(s.data());
        lengths.emplace_back(s.size());
      }
      return SetAttrStringList(attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

            op_->colocation_constraints.emplace(static_cast<const char*>(values[i]),
                                                lengths[i]);
          }
        } else {
          std::vector<tensorflow::StringPiece> v;
          v.reserve(num_values);
          for (int i = 0; i < num_values; ++i) {
            v.emplace_back(static_cast<const char*>(values[i]), lengths[i]);
          }
          op_->node_builder.Attr(attr_name, v);
        }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api.cc

          // String
          if (const int s_size = default_value.list().s_size()) {
            absl::InlinedVector<const void*, 4> values_vector;
            values_vector.reserve(s_size);
            absl::InlinedVector<size_t, 4> lengths_vector;
            lengths_vector.reserve(s_size);
            for (int i = 0; i < s_size; ++i) {
              const string& v = default_value.list().s(i);
              values_vector.push_back(v.data());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_debug.cc

                                             absl::Status* status) {
      std::vector<int64_t> shape;
      int rank = -1;
      *status = handle.NumDims(&rank);
      if (!status->ok()) {
        return shape;
      }
      shape.reserve(rank);
      for (int i = 0; i < rank; ++i) {
        int64_t dim;
        *status = handle.Dim(i, &dim);
        if (!status->ok()) {
          return shape;
        }
        shape.push_back(dim);
      }
      return shape;
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      std::vector<TensorHandlePtr> release_tensors() { return std::move(tensors_); }
    
      std::vector<TFE_TensorHandle*> tensors() const {
        std::vector<TFE_TensorHandle*> result;
        result.reserve(tensors_.size());
        for (const TensorHandlePtr& tensor : tensors_) {
          result.emplace_back(tensor.get());
        }
        return result;
      }
    
     private:
      ParallelTensor(const ParallelDevice& device,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_experimental.cc

        // Note that we take the address of the elements in `all_input_tensors`
        // below. Allocate enough space so that no reallocation happens, which will
        // make the pointers invalid.
        all_input_tensors.reserve(num_inputs);
        for (int i = 0; i < num_inputs; ++i) {
          if (input_tensors[i] == nullptr) continue;
          all_input_tensors.emplace_back();
          Tensor& input_tensor = all_input_tensors.back();
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
Back to top