Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for DataType (0.19 sec)

  1. tensorflow/c/eager/unified_api_testutil.h

      TFE_TensorHandle* input_eager =
          TestTensorHandleWithDims<T, datatype>(eager_ctx, data, dims, num_dims);
      *tensor =
          unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get()));
      return absl::OkStatus();
    }
    
    // Return a scalar tensor handle with given value.
    template <class T, TF_DataType datatype>
    Status TestScalarTensorHandle(AbstractContext* ctx, const T value,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
          unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value));
      TF_SetStatus(s, static_cast<TF_Code>(status.code()),
                   tsl::NullTerminatedMessage(status));
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/checkpoint_reader.cc

        TF_Status* out_status) const {
      Status status;
      if (reader_ != nullptr) {
        status = reader_->GetTensor(name, out_tensor);
      } else {
        tensorflow::DataType dtype;
        tensorflow::TensorShape shape;
        status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
        if (status.ok()) {
          out_tensor->reset(new Tensor(dtype, shape));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Aug 25 21:29:12 GMT 2023
    - 5.5K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_macros.h

    #else
    #define TF_CAPI_EXPORT __attribute__((visibility("default")))
    #endif  // TF_CAPI_WEAK
    #endif  // _WIN32
    #endif  // SWIG
    
    // TF_Bool is the C API typedef for unsigned char, while TF_BOOL is
    // the datatype for boolean tensors.
    #ifndef TF_Bool
    #define TF_Bool unsigned char
    #endif  // TF_Bool
    
    // Macro used to calculate struct size for maintaining ABI stability across
    // different struct implementations.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 04:44:45 GMT 2023
    - 1.6K bytes
    - Viewed (1)
  5. tensorflow/c/eager/immediate_execution_tensor_handle.cc

      if (!s.ok()) {
        device_name = "<error fetching device name>";
      }
      return absl::StrCat("TensorHandle(", value_string, ", shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()), ", device=\"",
                          device_name, "\")");
    }
    
    Status ImmediateExecutionTensorHandle::SummarizeValue(
        std::string& summary) const {
      Status status;
      AbstractTensorPtr resolved(
    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)
  6. tensorflow/c/eager/c_api_unified_experimental.h

    #ifndef TENSORFLOW_C_EAGER_C_API_UNIFIED_EXPERIMENTAL_H_
    #define TENSORFLOW_C_EAGER_C_API_UNIFIED_EXPERIMENTAL_H_
    
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/c/tf_datatype.h"
    #include "tensorflow/c/tf_status.h"
    
    #ifdef __cplusplus
    extern "C" {
    #endif
    
    // =============================================================================
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:00 GMT 2021
    - 7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/abstract_operation.h

      virtual Status SetAttrFloat(const char* attr_name, float value) = 0;
      virtual Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
      virtual Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                  const int num_dims) = 0;
      virtual Status SetAttrShape(const char* attr_name,
    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/c_api_test_util.h

    // Return a tensor handle with given type, values and dimensions.
    template <class T, TF_DataType datatype>
    TFE_TensorHandle* TestTensorHandleWithDims(TFE_Context* ctx, const T* data,
                                               const int64_t* dims, int num_dims) {
      TF_Status* status = TF_NewStatus();
      TF_Tensor* t = TFE_AllocateHostTensor(ctx, datatype, dims, num_dims, status);
      memcpy(TF_TensorData(t), data, TF_TensorByteSize(t));
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Jul 17 23:43:59 GMT 2023
    - 7.7K bytes
    - Viewed (0)
  9. tensorflow/c/c_test.c

    #include "tensorflow/c/kernels.h"
    
    // A create function. This will never actually get called in this test, it's
    // just nice to know that it compiles.
    void* create(TF_OpKernelConstruction* ctx) {
      TF_DataType type;
      TF_Status* s = TF_NewStatus();
      TF_OpKernelConstruction_GetAttrType(ctx, "foobar", &type, s);
      TF_DeleteStatus(s);
      return NULL;
    }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Apr 24 20:50:35 GMT 2024
    - 2.8K bytes
    - Viewed (0)
  10. tensorflow/c/eager/abstract_tensor_handle.cc

      if (!s.ok()) {
        shape_string = "<error computing shape>";
      } else {
        shape_string = shape.DebugString();
      }
      return absl::StrCat("TensorHandle(shape=", shape_string,
                          ", dtype=", DataType_Name(DataType()),
                          ", type=", FullType().DebugString(), ")");
    }
    
    Status AbstractTensorHandle::TensorHandleStatus() const {
      // Tensor handles in current runtime don't carry error info and this method
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
Back to top