- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for AbstractContext (0.63 sec)
-
tensorflow/c/eager/unified_api_testutil.h
namespace tensorflow { // Builds and returns a `TracingContext` using the default tracing impl. AbstractContext* BuildFunction(const char* fn_name); // Creates parameters (placeholders) in the tracing `ctx` using the shape and // dtype of `inputs`. Status CreateParamsForInputs(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
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.cc
: forward_outputs_(f_outputs) { for (auto output : forward_outputs_) { if (output) { output->Ref(); } } } Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { AbstractTensorHandle* upstream_grad = grad_outputs[0];
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/experimental/gradients/nn_grad_test.cc
namespace internal { namespace { using tensorflow::TF_StatusPtr; Status ReluModel(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { return ops::Relu(ctx, inputs[0], &outputs[0], "Relu"); } Status SparseSoftmaxCrossEntropyWithLogitsModel( AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs,
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/experimental/gradients/custom_gradient_test.cc
Status s = StatusFromTF_Status(status.get()); CHECK_EQ(errors::OK, s.code()) << s.message(); } }; class PassThroughGradientFunction : public GradientFunction { public: Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override { CHECK_EQ(grad_outputs.size(), 1);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
} TEST_P(CppGradients, TestSetAttrString) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); AbstractContextPtr ctx; { AbstractContext* ctx_raw = nullptr; Status s = BuildImmediateExecutionContext(std::get<1>(GetParam()), &ctx_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); ctx.reset(ctx_raw); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
using tensorflow::ops::Mul; using tensorflow::ops::Neg; using tensorflow::ops::OnesLike; using tensorflow::ops::SqrtGrad; namespace tensorflow { namespace gradients { namespace { static Status SafeConj(AbstractContext* ctx, AbstractTensorHandle* const input, AbstractTensorHandle** output, const char* name) { auto dtype = input->DataType(); if (DataTypeIsFloating(BaseType(dtype)) ||
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/experimental/gradients/array_grad_test.cc
#include "tensorflow/core/platform/test.h" namespace tensorflow { namespace gradients { namespace internal { namespace { using tensorflow::TF_StatusPtr; Status IdentityNModel(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { std::vector<AbstractTensorHandle*> temp_outputs(2); TF_RETURN_IF_ERROR(
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_test.cc
public: bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; } bool UseFunction() const { return std::get<2>(GetParam()); } }; // Checks that inputs[0] is a scalar. Status TestScalarShape(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle*> outputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(inputs[0]->Shape(&shape));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.cc
#include "tensorflow/core/platform/errors.h" namespace tensorflow { AbstractContext* BuildFunction(const char* fn_name) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TF_ExecutionContext* graph_ctx = TF_CreateFunction(fn_name, status.get()); return unwrap(graph_ctx); } Status CreateParamsForInputs(AbstractContext* ctx,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
#include "tensorflow/core/platform/test.h" namespace tensorflow { namespace gradients { namespace internal { void CompareNumericalAndAutodiffGradients( Model model, Model grad_model, AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> inputs, bool use_function, double abs_error) { auto num_inputs = inputs.size(); std::vector<AbstractTensorHandle*> outputs(num_inputs);
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0)