- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 32 for Down (0.13 sec)
-
tensorflow/c/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
if (!t) { return tensorflow::errors::InvalidArgument( "Unable to cast input to GraphTensor"); } inputs_.push_back(t->output_); *output = tensorflow::down_cast<TracingTensorHandle*>(outputs[0]); return absl::OkStatus(); } Status Finalize(OutputList* outputs, AbstractFunction** f) override { std::vector<TF_Output> graph_outputs;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/c_api_experimental.h
// Checks the health of collective ops peers. Explicit health check is needed in // multi worker collective ops to detect failures in the cluster. If a peer is // down, collective ops may hang. TF_CAPI_EXPORT extern void TFE_CollectiveOpsCheckPeerHealth( TFE_Context* ctx, const char* task, int64_t timeout_in_ms, TF_Status* status);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_test.cc
TF_AbstractFunction* func = TF_FinalizeFunction(graph_ctx, add_outputs, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); // Note: TF_OutputList does not own the underlying AbstractTensors, those // need to be deleted explicitly. TF_DeleteAbstractTensor(TF_OutputListGet(add_outputs, 0)); // Build eager context. TFE_ContextOptions* opts = TFE_NewContextOptions();
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 19 21:44:52 GMT 2023 - 39.1K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
&Deallocator, &deallocator_called); TF_Tensor* o = TF_TensorMaybeMove(t); ASSERT_TRUE(o == nullptr); // It is unsafe to move memory TF might not own. TF_DeleteTensor(t); EXPECT_TRUE(deallocator_called); } TEST(CAPI, LibraryLoadFunctions) { // TODO(b/73318067): Fix linking for the GPU test generated by the
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) -
tensorflow/c/eager/c_api_experimental.h
// The config key must be set before getting its value. Getting value of // non-existing config keys will result in errors. // If `timeout_in_ms=0`, this call will block until the key-value is set or the // worker shuts down. TF_CAPI_EXPORT extern void TFE_GetConfigKeyValue(TFE_Context* ctx, const char* key, int64_t timeout_in_ms,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/c_api.cc
tensorflow::TensorProto from_tensor_proto; status->status = BufferToMessage(from, &from_tensor_proto); if (!status->status.ok()) { return; } status->status = tensorflow::down_cast<tensorflow::TensorInterface*>(to->tensor) ->FromProto(from_tensor_proto); } // --------------------------------------------------------------------------
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
tensorflow::unwrap(h); // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here. if (tensorflow::CustomDeviceTensorHandle::classof(unwrapped_handle)) { return tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle) ->DevicePointer(); } // TODO(b/175427838): It would be nice to be able to use tensorflow::isa here.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/tape.h
template <typename BackwardFunction, typename TapeTensor> struct OpTapeEntry { string op_type; std::vector<TapeTensor> output_tensor_info; std::vector<int64_t> input_tensor_id; // TODO(apassos) consider narrowing down this interface. BackwardFunction* backward_function; // Should be called before deleting the backward function. TODO(apassos) use // unique_ptrs to ensure this happens.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/eager/abstract_operation.h
// Release any underlying resources, including the interface object. // // WARNING: The destructor of this class is marked as protected to disallow // clients from directly destroying this object since it may manage it's own // lifetime through ref counting. Thus this must be allocated on the heap and // clients MUST call Release() in order to destroy an instance of this class. virtual void Release() = 0;
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)