Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for tensor_id (0.18 sec)

  1. tensorflow/c/eager/tape.h

        std::vector<int64_t> unneeded_gradients;
        for (int i = 0, end = trace.input_tensor_id.size(); i < end; i++) {
          const auto& in_tensor_id = trace.input_tensor_id[i];
          if (tensor_tape_.find(in_tensor_id) == tensor_tape_.end() &&
              sources_set.find(in_tensor_id) == sources_set.end()) {
            unneeded_gradients.push_back(i);
          }
        }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  2. tensorflow/c/c_api_experimental.h

                                                           TF_Status* status);
    
    // On success, enqueues `tensor` into a TF-managed FifoQueue given by
    // `tensor_id`, associated with `session`. There must be a graph node named
    // "fifo_queue_enqueue_<tensor_id>", to be executed by this API call. It reads
    // from a placeholder node "arg_tensor_enqueue_<tensor_id>".
    //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

      }
      return ret;
    }
    
    TF_Tensor* TF_DequeueNamedTensor(TF_Session* session, int tensor_id,
                                     TF_Status* status) {
      assert(session);
      {
        tensorflow::mutex_lock c(session->graph->mu);
        VLOG(1) << "Dequeuing named tensor with id " << tensor_id
                << ", with input graph: "
                << session->graph->graph.ToGraphDefDebug().DebugString();
      }
    
    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)
  4. tensorflow/c/c_api_internal.h

      std::atomic<bool> extend_before_run;
    };
    
    struct TF_ImportGraphDefOptions {
      tensorflow::ImportGraphDefOptions opts;
    
      // Backing memory for TensorId fields in opts.
      // TODO(skyewm): it'd be better if ImportGraphDefOptions owned this.
      std::vector<tensorflow::string> tensor_id_data;
    };
    
    struct TF_ImportGraphDefResults {
      std::vector<TF_Output> return_tensors;
      std::vector<TF_Operation*> return_nodes;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 00:49:12 GMT 2023
    - 7.6K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

                                                 int src_index, TF_Output dst) {
      opts->tensor_id_data.push_back(src_name);
      const string& src_name_str = opts->tensor_id_data.back();
      // We don't need to store dst's name in tensor_id_data, since `dst` must
      // outlive the ImportGraphDef call.
      opts->opts.input_map[TensorId(src_name_str, src_index)] = ToTensorId(dst);
    }
    
    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)
  6. tensorflow/c/c_api_test.cc

    #include "tensorflow/core/framework/tensor.h"
    #include "tensorflow/core/framework/tensor.pb.h"
    #include "tensorflow/core/framework/tensor_shape.pb.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/graph/tensor_id.h"
    #include "tensorflow/core/lib/core/status_test_util.h"
    #include "tensorflow/core/lib/io/path.h"
    #include "tensorflow/core/platform/path.h"
    #include "tensorflow/core/platform/protobuf.h"
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  7. tensorflow/c/eager/gradients.cc

      Status BuildOnesLike(const TapeTensor& t,
                           AbstractTensorHandle** result) const override;
    
      // Looks up the ID of a Gradient.
      int64_t TensorId(AbstractTensorHandle* tensor) const override;
    
      // Converts a Gradient to a TapeTensor.
      TapeTensor TapeTensorFromGradient(AbstractTensorHandle* g) const override;
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
Back to top