Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for handle_op (0.1 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      ParallelDevice parallel_device(std::move(devices));
      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_experimental_test.cc

        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_Execute(op, &handle_1, &num_retvals, status);
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_DeleteOp(op);
    
        ASSERT_EQ(1, num_retvals);
        EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(handle_1));
        EXPECT_EQ(0, TFE_TensorHandleNumDims(handle_1, status));
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 03:14:26 UTC 2023
    - 31.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/gradients.cc

    }
    
    TapeTensor::TapeTensor(AbstractTensorHandle* handle) : handle_(handle) {
      handle_->Ref();
    }
    TapeTensor::TapeTensor(const TapeTensor& other) {
      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();
    }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top