- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 68 for returns (0.24 sec)
-
tensorflow/c/checkpoint_reader.h
bool HasTensor(const string& name) const; const string DebugString() const; // Returns a map from variable names to their shapes. Slices of a partitioned // tensor are combined into a single entry. const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const; // Returns a map from variable names to their data types. Slices of a // partitioned tensor are combined into a single entry.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 12 08:49:52 GMT 2023 - 3.1K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.h
TF_ExecutionContextGetTFEContext(wrap(ctx), status.get()); TF_RETURN_IF_ERROR(StatusFromTF_Status(status.get())); 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.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
"SparseSoftmaxCrossEntropyWithLogits")); // `gradient_checker` only works with model that returns only 1 tensor. // Although, `ops::SparseSoftmaxCrossEntropyWithLogits` returns 2 tensors, the // second tensor isn't needed for computing gradient so we could safely drop // it. outputs[0] = loss; backprop->Unref(); return absl::OkStatus(); } Status BiasAddModel(AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
void* device_info) = nullptr; // Pins the op to `device` based on inputs to `op`. Returns true // signifying to pin to the current custom device. Returns false // to pin to the physical device. // // This function is guaranteed to be called only when all of the custom-device // inputs are on this device.
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/experimental/next_pluggable_device/c_api.h
// Returns a not owning pointer to the coordination service agent, which is // opaque to plugin. Plugin OpKernels need to use the accompanying C APIs to // access coordination service functionalities. TF_CAPI_EXPORT extern TF_CoordinationServiceAgent* TF_GetCoordinationServiceAgent(TF_OpKernelContext* ctx); // Returns true if the coordination service agent has been initialized.
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/graph_function.h
~GraphFunction() override; // GraphFunction maybe stay alive for the duration of the returned // FunctionDef. Status GetFunctionDef(const FunctionDef** fdef) override; // Returns a shared reference to the wrapped function. absl::StatusOr<core::RefCountPtr<FunctionRecord>> GetFunctionRecord() override { return func_record_.GetNewRef(); } // For LLVM style RTTI.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
// and hence expect the input to be forwarded to the return tensor. EXPECT_EQ(orig_ptr, TF_TensorData(t)); } else { // In sync mode, forwarding can't really happen since the client code will // have a reference count on the input tensor while the kernel is being // executed and thus it cannot be re-used for the return tensor. EXPECT_NE(orig_ptr, TF_TensorData(t)); } }
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) -
tensorflow/c/eager/gradients.cc
} 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 : public eager::VSpace<AbstractTensorHandle, GradientFunction, TapeTensor> {
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) -
tensorflow/c/eager/immediate_execution_context.h
// Find and return a added function by its name. virtual const FunctionDef* FindFunctionDef(const string& name) const = 0; // Find and return a function record added by its name. virtual core::RefCountPtr<FunctionRecord> FindRecord( const string& name) const = 0; // Return the ParsedName of Host CPU device.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/abstract_function.h
explicit AbstractFunction(AbstractFunctionKind kind) : kind_(kind) {} public: // Returns which subclass is this instance of. AbstractFunctionKind getKind() const { return kind_; } // Returns the AbstractFunction as a FunctionDef. virtual Status GetFunctionDef(const FunctionDef**) = 0; // Returns a shared reference to the wrapped function. virtual absl::StatusOr<core::RefCountPtr<FunctionRecord>>
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.9K bytes - Viewed (0)