Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for References (0.25 sec)

  1. 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)
  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/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)
  6. 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)
  7. 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)
  8. tensorflow/c/c_api_function_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      DefineT(-1, {}, {{split, 0}, {split, 2}}, {{add, 0}}, {}, true);
      EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(s_));
      EXPECT_EQ(string("When `num_opers` is set to -1, nodes referenced in "
                       "`inputs` must have a single output. Node split3 has "
                       "3 outputs. Encountered while creating function 'MyFunc'"),
                string(TF_Message(s_)));
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top