Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for reference (0.23 sec)

  1. tensorflow/c/c_api.cc

    }
    
    namespace {
    
    // Helper method that creates a shape handle for a shape described by dims.
    tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims(
        tensorflow::shape_inference::InferenceContext* ic, int num_dims,
        const int64_t* dims) {
      if (num_dims != -1) {
        std::vector<tensorflow::shape_inference::DimensionHandle> dim_vec;
        dim_vec.reserve(num_dims);
        for (int i = 0; i < num_dims; ++i) {
    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/c_api_function.cc

          const auto& iter = input_nodes.find(node);
          if (iter == input_nodes.end()) {
            // This node is not referenced in inputs. Add it to the body.
            body_nodes->push_back(node);
          } else {
            // This node is referenced in inputs. Currently, we place an
            // artificial restriction and require that when num_opers=-1, such
            // nodes must have a single output.
    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)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

      return result;
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, indicating how
    // ParallelTensors wrapped in TFE_TensorHandles should be cleaned up once their
    // reference counts drop to zero.
    void ParallelTensorDeallocator(void* data) {
      delete reinterpret_cast<ParallelTensor*>(data);
    }
    
    // Used as an argument to TFE_NewCustomDeviceTensorHandle, for computing the
    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

        std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> await_status(
            TF_NewStatus(), TF_DeleteStatus);
        // Wait until all pending nodes have completed since they may have a
        // reference to default_cancellation_manager_. We ignore the status return
        // since we already have a bad status to propagate.
        TFE_ContextAsyncWait(context, await_status.get());
    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/eager/dlpack.cc

    // original framework of destruction, and this context will be deleted also.
    struct TfDlManagedTensorCtx {
      TensorReference reference;
      std::vector<int64_t> shape;
      std::vector<int64_t> strides;
      DLManagedTensor tensor;
    
      explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {}
    };
    
    // Gets tensor from eager tensor handle.
    const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_experimental.cc

      using tensorflow::NodeDef;
      using tensorflow::OpRegistrationData;
      using tensorflow::Tensor;
      using tensorflow::shape_inference::DimensionHandle;
      using tensorflow::shape_inference::InferenceContext;
      using tensorflow::shape_inference::ShapeAndType;
      using tensorflow::shape_inference::ShapeHandle;
    
      const int num_inputs = input_shapes->num_items;
      NodeDef node_def;
    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/modular_filesystem_test.cc

        s = fs->CreateDir(root_dir_);
        if (!s.ok()) {
          GTEST_SKIP() << "Cannot create working directory: " << s;
        }
      }
    
      // Converts path reference to URI reference.
      //
      // If URI scheme is empty, URI reference is `path` relative to current test
      // root directory. Otherwise, we need to add the `<scheme>://` in front of
      // this path.
      //
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 27 20:25:58 GMT 2022
    - 71K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_test.cc

          EXPECT_EQ(orig_ptr, TF_TensorData(t));
        } else {
          // In sync mode, forwarding can't really happen since the client code will
          // have a reference count on the input tensor while the kernel is being
          // executed and thus it cannot be re-used for the return tensor.
          EXPECT_NE(orig_ptr, TF_TensorData(t));
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  9. tensorflow/c/c_api_test.cc

          .Attr("v: " #type)                                      \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape); \
      REGISTER_OP("CApiAttributesTestOpList" #type)               \
          .Attr("v: list(" #type ")")                             \
          .SetShapeFn(tensorflow::shape_inference::UnknownShape)
    ATTR_TEST_REGISTER_OP(string);
    ATTR_TEST_REGISTER_OP(int);
    ATTR_TEST_REGISTER_OP(float);
    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)
  10. tensorflow/c/eager/gradients.cc

      ~TapeVSpace() override {}
    
      // Returns the number of elements in the gradient tensor.
      int64_t NumElements(AbstractTensorHandle* tensor) const override;
    
      // Consumes references to the tensors in the gradient_tensors list and returns
      // a tensor with the result.
      AbstractTensorHandle* AggregateGradients(
          gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) 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