Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 83 for vendor (0.17 sec)

  1. tensorflow/c/eager/gradients.cc

        absl::Span<const AbstractTensorHandle* const> tensors) const {
      std::vector<int64_t> tensor_ids(tensors.size());
      std::vector<tensorflow::DataType> tensor_dtypes(tensors.size());
      for (int i = 0; i < tensors.size(); i++) {
        tensor_ids[i] = ToId(tensors[i]);
        tensor_dtypes[i] = tensors[i]->DataType();
      }
      return GradientTape::ShouldRecord(tensor_ids, tensor_dtypes);
    }
    void Tape::DeleteTrace(const AbstractTensorHandle* t) {
    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/eager/tfe_tensor_debug_info_internal.h

    #ifndef TENSORFLOW_C_EAGER_TFE_TENSOR_DEBUG_INFO_INTERNAL_H_
    #define TENSORFLOW_C_EAGER_TFE_TENSOR_DEBUG_INFO_INTERNAL_H_
    
    #include <vector>
    
    #include "tensorflow/core/platform/types.h"
    
    struct TFE_TensorDebugInfo {
      explicit TFE_TensorDebugInfo(const std::vector<int64_t>& dims)
          : dev_dims(dims) {}
    
      // Fully-padded, minor-to-major.
      std::vector<int64_t> dev_dims;
    };
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 1.1K bytes
    - Viewed (0)
  3. tensorflow/c/c_test_util.cc

                              const char* name) {
      unique_tensor_ptr tensor(Int32Tensor(v), TF_DeleteTensor);
      return Const(tensor.get(), graph, s, name);
    }
    
    TF_Operation* ScalarConst(double v, TF_Graph* graph, TF_Status* s,
                              const char* name) {
      unique_tensor_ptr tensor(DoubleTensor(v), TF_DeleteTensor);
      return Const(tensor.get(), graph, s, name);
    }
    
    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)
  4. tensorflow/c/eager/dlpack.cc

      DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor;
      dlm_tensor->manager_ctx = tf_dlm_tensor_ctx;
      dlm_tensor->deleter = &DLManagedTensorDeleter;
      dlm_tensor->dl_tensor.device = tf_dlm_context;
      int ndim = tensor->dims();
      dlm_tensor->dl_tensor.ndim = ndim;
      dlm_tensor->dl_tensor.data = tf_dlm_data;
      dlm_tensor->dl_tensor.dtype = tf_dlm_type;
    
    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)
  5. tensorflow/c/experimental/gradients/nn_grad.cc

    #include "tensorflow/c/experimental/gradients/nn_grad.h"
    
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/eager/immediate_execution_context.h"
    #include "tensorflow/c/eager/immediate_execution_tensor_handle.h"
    #include "tensorflow/c/experimental/ops/array_ops.h"
    #include "tensorflow/c/experimental/ops/math_ops.h"
    #include "tensorflow/c/experimental/ops/nn_ops.h"
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/dlpack_test.cc

    #include "tensorflow/c/eager/dlpack.h"
    
    #include <vector>
    
    #include "absl/strings/str_join.h"
    #include "include/dlpack/dlpack.h"  // from @dlpack
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace {
    
    void TestHandleFromDLPack(TF_Status* status, TFE_Context* ctx,
                              std::vector<int64_t> shape,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Jun 30 03:04:46 GMT 2023
    - 4.4K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_operation.h

    #define TENSORFLOW_C_EAGER_ABSTRACT_OPERATION_H_
    
    #include <memory>
    
    #include "absl/types/span.h"
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/tensor_interface.h"
    #include "tensorflow/core/framework/tensor_shape.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    
    // Abstract interface to an operation.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  8. tensorflow/c/eager/immediate_execution_tensor_handle.cc

    See the License for the specific language governing permissions and
    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/eager/immediate_execution_tensor_handle.h"
    
    namespace tensorflow {
    
    std::string ImmediateExecutionTensorHandle::DebugString() const {
      PartialTensorShape shape;
      std::string shape_string;
      if (Shape(&shape).ok()) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 2.1K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/array_grad_test.cc

                          absl::Span<AbstractTensorHandle*> outputs) {
      std::vector<AbstractTensorHandle*> temp_outputs(2);
      TF_RETURN_IF_ERROR(
          ops::IdentityN(ctx, inputs, absl::MakeSpan(temp_outputs), "IdentityN"));
      // Although, `ops::IdentityN` returns 2 tensors, the first tensor isn't needed
      // for computing gradient so we could safely drop it.
      outputs[0] = temp_outputs[1];
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  10. tensorflow/c/eager/abstract_tensor_handle.h

      // Returns the status of the tensor handle. If it is a tfrt::TensorHandle,
      // the tensor handle can be an error and return non-OK status.
      virtual tensorflow::Status TensorHandleStatus() const;
    
      // Returns tensor shape. If tensor has unknown rank, shape remains untouched.
      virtual tensorflow::Status Shape(
          tensorflow::PartialTensorShape* shape) const = 0;
    
      // Returns tensor (full) type.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 03 00:30:36 GMT 2023
    - 3K bytes
    - Viewed (0)
Back to top