Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 31 for TFE_TensorHandle (0.27 sec)

  1. tensorflow/c/eager/custom_device_testutil.h

    void AllocateLoggingDevice(const char* name, bool* arrived_flag,
                               bool* executed_flag, TFE_CustomDevice** device,
                               void** device_info);
    TFE_TensorHandle* UnpackTensorHandle(TFE_TensorHandle* logged_tensor_handle,
                                         TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 27 23:39:24 UTC 2020
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_eager.cc

      }
      return wrap(static_cast<AbstractContext*>(unwrap(c_ctx)));
    }
    
    TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t,
                                                              TF_Status* s) {
      return wrap(static_cast<AbstractTensorHandle*>(unwrap(t)));
    }
    
    TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                      TF_Status* s) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jun 25 04:40:46 UTC 2020
    - 3.2K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.h

      // component device.
      absl::Status SummarizeValue(std::string& summary);
    
      std::vector<TensorHandlePtr> release_tensors() { return std::move(tensors_); }
    
      std::vector<TFE_TensorHandle*> tensors() const {
        std::vector<TFE_TensorHandle*> result;
        result.reserve(tensors_.size());
        for (const TensorHandlePtr& tensor : tensors_) {
          result.emplace_back(tensor.get());
        }
        return result;
      }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental.h

    // tensors and how to get value out of eager abstract tensors.
    TF_AbstractTensor* TF_CreateAbstractTensorFromEagerTensor(TFE_TensorHandle* t,
                                                              TF_Status* s);
    TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                      TF_Status* s);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sun Oct 24 11:27:00 UTC 2021
    - 7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/c_api_experimental.cc

        return nullptr;
      }
    
      return new TF_Tensor{t};
    }
    
    TFE_TensorHandle* TFE_NewTensorHandleFromTensor(TFE_Context* ctx, TF_Tensor* t,
                                                    TF_Status* status) {
      return tensorflow::wrap(
          tensorflow::unwrap(ctx)->CreateLocalHandle(t->tensor));
    }
    
    TFE_TensorHandle* TFE_CreatePackedTensorHandle(TFE_Context* ctx,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 35.9K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_debug.cc

          return shape;
        }
        shape.push_back(dim);
      }
      return shape;
    }
    
    }  // namespace
    
    extern "C" {
    
    TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(
        TFE_TensorHandle* h, TF_Status* status) {
      tensorflow::TensorHandle* handle =
          TensorHandleFromInterface(tensorflow::unwrap(h));
      const tensorflow::Tensor* tensor;
      status->status = handle->Tensor(&tensor);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  7. tensorflow/c/eager/dlpack_test.cc

      dltensor_in->device = {kDLCPU, 0};
      dltensor_in->ndim = static_cast<int32_t>(shape.size());
      dltensor_in->dtype = {kDLFloat, 32, 1};
      dltensor_in->shape = shape.data();
      dltensor_in->strides = strides.data();
      TFE_TensorHandle* handle = TFE_HandleFromDLPack(&dlm_in, status, ctx);
      ASSERT_NE(handle, nullptr)
          << TF_Message(status) << " (shape=[" << absl::StrJoin(shape, ",")
          << "], strides=[" << absl::StrJoin(strides, ",") << "])";
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Jun 30 03:04:46 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc

      TensorHandlePtr value_two(FloatTensorHandle(-2., status.get()));
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      std::array<TFE_TensorHandle*, 2> in_components{value_one.get(),
                                                     value_two.get()};
      TensorHandlePtr combined_value = CreatePerDeviceValues(
          context.get(), in_components, device_name, status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Jul 10 07:18:05 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  9. tensorflow/c/c_api_experimental.h

    // in tests only.)
    TF_CAPI_EXPORT int TF_PickUnusedPortOrDie(void);
    
    // Fast path method that makes constructing a single scalar tensor require less
    // overhead and copies.
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromScalar(
        TF_DataType data_type, void* data, size_t len, TF_Status* status);
    
    // Specify the server_def that enables collective ops.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_experimental_test.cc

      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      TFE_OpSetAttrType(fill_op, "T", TF_FLOAT);
      TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32);
    
      float five = 5.0;
      TFE_TensorHandle* scalar = TestScalarTensorHandle(tfe_context_, five);
      TF_Tensor* scalarTensor = TFE_TensorHandleResolve(scalar, status_);
      CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);
      CheckOutputShapes(fill_op,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Jan 17 22:27:52 UTC 2023
    - 13.1K bytes
    - Viewed (0)
Back to top