Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 45 for DataType (0.15 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/dlpack.cc

          break;
        case TF_DataType::TF_INT8:
        case TF_DataType::TF_INT16:
        case TF_DataType::TF_INT32:
        case TF_DataType::TF_INT64:
          dtype.code = DLDataTypeCode::kDLInt;
          break;
        case TF_DataType::TF_UINT8:
        case TF_DataType::TF_UINT16:
        case TF_DataType::TF_UINT32:
        case TF_DataType::TF_UINT64:
          dtype.code = DLDataTypeCode::kDLUInt;
          break;
    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)
  3. tensorflow/c/eager/abstract_op_attrs.h

      virtual bool GetBool(absl::string_view attr_name, bool* result) const = 0;
      virtual bool GetType(absl::string_view attr_name, DataType* result) const = 0;
      virtual Status GetTypeList(
          absl::string_view attr_name,
          absl::InlinedVector<DataType, 4>* type_list) const = 0;
    
     private:
      const AbstractOpAttrsKind kind_;
    };
    
    }  // namespace tensorflow
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed May 26 22:20:27 GMT 2021
    - 2K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    }
    Status TapeOperation::SetAttrTypeList(const char* attr_name,
                                          const DataType* values, int num_values) {
      forward_op_.attrs.Set(attr_name,
                            gtl::ArraySlice<const DataType>(values, num_values));
      return parent_op_->SetAttrTypeList(attr_name, values, num_values);
    }
    Status TapeOperation::SetAttrBoolList(const char* attr_name,
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  5. 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)
  6. 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)
  7. tensorflow/c/c_api_test.cc

      EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT));
      EXPECT_EQ(TF_DOUBLE, static_cast<TF_DataType>(tensorflow::DT_DOUBLE));
      EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32));
      EXPECT_EQ(TF_UINT8, static_cast<TF_DataType>(tensorflow::DT_UINT8));
      EXPECT_EQ(TF_INT16, static_cast<TF_DataType>(tensorflow::DT_INT16));
      EXPECT_EQ(TF_INT8, static_cast<TF_DataType>(tensorflow::DT_INT8));
    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)
  8. tensorflow/c/eager/gradients.cc

      handle_ = other.handle_;
      handle_->Ref();
    }
    TapeTensor::~TapeTensor() { handle_->Unref(); }
    
    int64_t TapeTensor::GetID() const { return ToId(handle_); }
    
    tensorflow::DataType TapeTensor::GetDType() const {
      return handle_->DataType();
    }
    AbstractTensorHandle* TapeTensor::GetHandle() const { return handle_; }
    
    AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; }
    
    class TapeVSpace
    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)
  9. 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)
  10. tensorflow/c/BUILD

            ],
        }),
    )
    
    cc_library(
        name = "tf_datatype_hdrs",
        hdrs = ["tf_datatype.h"],
        visibility = ["//tensorflow:internal"],
        deps = [
            ":c_api_macros_hdrs",
        ],
    )
    
    cc_library(
        name = "tf_datatype",
        srcs = ["tf_datatype.cc"],
        hdrs = ["tf_datatype.h"],
        copts = tf_copts(),
        visibility = ["//visibility:public"],
        deps = [
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 27 18:00:18 GMT 2024
    - 30.3K bytes
    - Viewed (0)
Back to top