- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for Implicit (0.22 sec)
-
tensorflow/c/c_api_test.cc
TF_DeleteBuffer(api_def_buf); TF_DeleteApiDefMap(api_def_map); TF_DeleteBuffer(op_list_buf); } class DummyKernel : public tensorflow::OpKernel { public: explicit DummyKernel(tensorflow::OpKernelConstruction* context) : OpKernel(context) {} void Compute(tensorflow::OpKernelContext* context) override {} }; // Test we can query kernels
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/experimental/gradients/nn_grad.cc
using tensorflow::ops::Mul; using tensorflow::ops::ReluGrad; namespace tensorflow { namespace gradients { namespace { class ReluGradientFunction : public GradientFunction { public: explicit ReluGradientFunction(vector<AbstractTensorHandle*> f_outputs) : forward_outputs_(f_outputs) { for (auto output : forward_outputs_) { if (output) { output->Ref(); } } }
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/eager/graph_function.h
namespace tensorflow { namespace tracing { namespace graph { using tensorflow::AbstractFunction; // Thin wrapper around a FunctionDef. class GraphFunction : public AbstractFunction { public: explicit GraphFunction(FunctionDef fdef); ~GraphFunction() override; // GraphFunction maybe stay alive for the duration of the returned // FunctionDef. Status GetFunctionDef(const FunctionDef** fdef) override;
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/dlpack.cc
struct TfDlManagedTensorCtx { TensorReference reference; std::vector<int64_t> shape; std::vector<int64_t> strides; DLManagedTensor tensor; explicit TfDlManagedTensorCtx(const TensorReference& ref) : reference(ref) {} }; // Gets tensor from eager tensor handle. const Tensor* GetTensorFromHandle(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) {
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) -
tensorflow/c/eager/gradients.cc
AbstractTensorHandle* TapeTensor::ZerosLike() const { return nullptr; } class TapeVSpace : public eager::VSpace<AbstractTensorHandle, GradientFunction, TapeTensor> { public: explicit TapeVSpace(AbstractContext* ctx) : ctx_(ctx) {} ~TapeVSpace() override {} // Returns the number of elements in the gradient tensor. int64_t NumElements(AbstractTensorHandle* tensor) const override;
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/experimental/gradients/math_grad.cc
grad_inputs[0]->Ref(); grad_inputs[1]->Ref(); return absl::OkStatus(); } ~AddGradientFunction() override {} }; class ExpGradientFunction : public GradientFunction { public: explicit ExpGradientFunction(AbstractTensorHandle* exp) : exp_(exp) { exp->Ref(); } Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// otherwise be the name of this custom device. Ops are placed onto a custom // device if any of their inputs are on that custom device, but custom devices // are free to set a bad status in order to require explicit placement. void (*execute)(const TFE_Op* op, int* num_outputs, TFE_TensorHandle** outputs, TF_Status* s, void* device_info); // Method to delete a 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/eager/c_api_unified_experimental_graph.cc
// GraphTensor wraps a `TF_Output`, i.e. a pointer to TF_Operation and the index // into the list of outputs for the operation. class GraphTensor : public TracingTensorHandle { public: explicit GraphTensor(TF_Output output, TF_Graph* graph) : TracingTensorHandle(kGraph), output_(output), graph_(graph) {} tensorflow::DataType DataType() const override {
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) -
RELEASE.md
and works in the DTensor training use case. * Improved masking support for [`tf.keras.layers.MultiHeadAttention`](https://www.tensorflow.org/api_docs/python/tf/keras/layers/MultiHeadAttention). * Implicit masks for `query`, `key` and `value` inputs will automatically be used to compute a correct attention mask for the layer. These padding masks will be combined with any
Plain Text - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Mon Apr 29 19:17:57 GMT 2024 - 727.7K bytes - Viewed (8) -
tensorflow/c/eager/abstract_function.h
// representation between various supported formats e.g. FunctionDef and Mlir // function. class AbstractFunction : public core::RefCounted { protected: enum AbstractFunctionKind { kGraph, kMlir }; explicit AbstractFunction(AbstractFunctionKind kind) : kind_(kind) {} public: // Returns which subclass is this instance of. AbstractFunctionKind getKind() const { return kind_; }
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)