Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 85 for dtype (0.14 sec)

  1. tensorflow/c/eager/dlpack_test.cc

      EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type);
      EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id);
      EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim);
      EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code);
      EXPECT_EQ(dltensor_out->dtype.bits, dltensor_in->dtype.bits);
      EXPECT_EQ(dltensor_out->dtype.lanes, dltensor_in->dtype.lanes);
      for (int i = 0; i < dltensor_in->ndim; ++i) {
    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)
  2. tensorflow/c/checkpoint_reader.cc

      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));
          status = v2_reader_->Lookup(name, out_tensor->get());
          if (!status.ok()) out_tensor->reset();
        }
      }
    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)
  3. tensorflow/c/eager/dlpack.cc

      owner->reference.Unref();
      delete owner;
    }
    
    // Converts TF_DATAType to DLPack data type.
    DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) {
      DLDataType dtype;
      dtype.lanes = 1;
      dtype.bits = TF_DataTypeSize(data_type) * 8;
      switch (data_type) {
        case TF_DataType::TF_BOOL:
          dtype.code = DLDataTypeCode::kDLBool;
          break;
        case TF_DataType::TF_HALF:
        case TF_DataType::TF_FLOAT:
    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)
  4. tensorflow/c/eager/c_api_test.cc

                name_and_attrs.attr().find("dtype")->second.type());
      TF_DeleteBuffer(serialized_attr_values);
    
      TFE_Op* var_op_2 = TFE_NewOp(ctx, "VarHandleOp", status);
    
      string serialized_dtype;
      ASSERT_TRUE(name_and_attrs.attr().find("dtype")->second.SerializeToString(
          &serialized_dtype));
      TFE_OpSetAttrValueProto(
          var_op_2, "dtype",
    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)
  5. tensorflow/c/c_test_util.cc

      }
      return found_dtype && found_shape;
    }
    
    bool IsScalarConst(const tensorflow::NodeDef& node_def, int v) {
      if (node_def.op() != "Const" || node_def.name() != "scalar") {
        return false;
      }
      bool found_dtype = false;
      bool found_value = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "dtype") {
          if (attr.second.type() == tensorflow::DT_INT32) {
            found_dtype = true;
    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)
  6. tensorflow/c/eager/immediate_execution_tensor_handle.h

      virtual bool PreferCustomSummarizer() const { return false; }
    
      // Returns a string which summarizes the value of this TensorHandle, for
      // debugging. Does not include a shape or dtype.
      //
      // Included in the default implementation of DebugString.
      virtual Status SummarizeValue(std::string& summary) const;
    
      // For LLVM style RTTI.
      static bool classof(const AbstractTensorHandle* ptr) {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 10 21:56:24 GMT 2023
    - 4.3K bytes
    - Viewed (0)
  7. tensorflow/c/eager/unified_api_testutil.h

    namespace tensorflow {
    
    // Builds and returns a `TracingContext` using the default tracing impl.
    AbstractContext* BuildFunction(const char* fn_name);
    
    // Creates parameters (placeholders) in the tracing `ctx` using the shape and
    // dtype of `inputs`.
    Status CreateParamsForInputs(AbstractContext* ctx,
                                 absl::Span<AbstractTensorHandle* const> inputs,
                                 std::vector<AbstractTensorHandle*>* params);
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental.cc

                               TF_Status* s) {
      unwrap(o)->outputs.push_back(unwrap(tensor));
    }
    
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  9. tensorflow/c/eager/abstract_tensor_handle.h

          tensorflow::PartialTensorShape* shape) const = 0;
    
      // Returns tensor (full) type.
      // While there is no immediate plan to deprecate dtype and shape in favor
      // of only using full type type information, this is a future possibility.
      //
      // Note that map_dtype_to_child_of_tensor() from core/framework/types.h
      // can be used to set a FullTypeDef based on dtype in a derived class if
      // appropriate.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 03 00:30:36 GMT 2023
    - 3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/custom_device_test.cc

          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
      TFE_OpSetAttrString(op.get(), "container", "", 0);
      TFE_OpSetAttrString(op.get(), "shared_name", "", 0);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
Back to top