Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 16 for template (0.68 sec)

  1. tensorflow/c/c_api.cc

        input_list.emplace_back(&inputs[i].oper->node, inputs[i].index);
      }
      desc->node_builder.Input(input_list);
    }
    
    void TF_AddControlInput(TF_OperationDescription* desc, TF_Operation* input) {
      desc->node_builder.ControlInput(&input->node);
    }
    
    void TF_ColocateWith(TF_OperationDescription* desc, TF_Operation* op) {
      desc->colocation_constraints.emplace(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_testlib.cc

          TF_DeleteTensor);
      return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status));
    }
    
    // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle.
    template <std::size_t num_replicas>
    void ExtractPerDeviceValues(
        TFE_Context* context, TFE_TensorHandle* input,
        std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

        for (int i = 0; i < t->num_tensors(); ++i) {
          TensorHandlePtr this_output(
              TFE_TensorHandleCopySharingTensor(t->tensor(i), status));
          outputs.emplace_back(std::move(this_output));
          if (TF_GetCode(status) != TF_OK) return result;
        }
        result.emplace(std::move(outputs));
        return result;
      }
      std::vector<ParallelTensor*> parallel_inputs;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

        return;
      }
      unwrapped_results.resize(real_num_outputs);
      outputs->reserve(real_num_outputs);
      for (TFE_TensorHandle* unwrapped_result : unwrapped_results) {
        outputs->emplace_back(unwrapped_result);
      }
    }
    
    ParallelDevice::ParallelDevice(const std::vector<std::string>& devices,
                                   bool is_async, int in_flight_nodes_limit)
        : underlying_devices_(devices),
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  5. tensorflow/c/c_api_function.cc

                                        "Encountered while processing input ", i,
                                        " into function '", fn_name, "'");
    
        input_tensors->emplace_back(node, idx);
    
        const auto& iter = input_nodes->find(node);
        if (iter == input_nodes->end()) {
          input_nodes->insert({node, {idx}});
        } else {
          auto& indices = iter->second;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  6. tensorflow/c/c_api_experimental.cc

        // 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();
          status->status = TF_TensorToTensor(input_tensors[i], &input_tensor);
          if (!status->status.ok()) return;
          input_tensors_vector[i] = &input_tensor;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.cc

      lru_list_.push_front(key);
      lra_list_.push_front(key);
      new_entry->lru_iterator = lru_list_.begin();
      new_entry->lra_iterator = lra_list_.begin();
      new_entry->timestamp = timer_seconds_();
      block_map_.emplace(std::make_pair(key, new_entry));
      return new_entry;
    }
    
    // Remove blocks from the cache until we do not exceed our maximum size.
    void RamFileBlockCache::Trim() {
      while (!lru_list_.empty() && cache_size_ > max_bytes_) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 16 01:39:09 GMT 2020
    - 11.1K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc

          block_size, 2 * callers * block_size, 0, fetcher);
      std::vector<std::unique_ptr<Thread>> threads;
      threads.reserve(callers);
      for (int i = 0; i < callers; i++) {
        threads.emplace_back(
            Env::Default()->StartThread({}, "caller", [block_size, &cache, i]() {
              std::vector<char> out;
              TF_EXPECT_OK(
                  ReadCache(&cache, "a", i * block_size, block_size, &out));
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:57 GMT 2021
    - 23.2K bytes
    - Viewed (0)
  9. tensorflow/c/c_test_util.cc

      inputs_.clear();
      for (const auto& p : inputs) {
        inputs_.emplace_back(TF_Output{p.first, 0});
        input_values_.emplace_back(p.second);
      }
    }
    
    void CSession::SetOutputs(std::initializer_list<TF_Operation*> outputs) {
      ResetOutputValues();
      outputs_.clear();
      for (TF_Operation* o : outputs) {
        outputs_.emplace_back(TF_Output{o, 0});
      }
      output_values_.resize(outputs_.size());
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  10. tensorflow/c/eager/c_api_experimental_reader.cc

    ==============================================================================*/
    
    #include "tensorflow/c/eager/c_api_experimental_reader.h"
    
    #include "tensorflow/c/eager/tfe_monitoring_reader_internal.h"
    
    template <typename... LabelType>
    int64_t TFE_MonitoringCounterReader::Read(const LabelType&... labels) {
      return counter->Read(labels...);
    }
    
    TFE_MonitoringCounterReader* TFE_MonitoringNewCounterReader(const char* name) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 20 03:14:47 GMT 2023
    - 1.4K bytes
    - Viewed (0)
Back to top