Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 80 for Op (0.13 sec)

  1. tensorflow/c/eager/c_api_experimental_test.cc

        int num_retvals = 1;
        TF_Status* status = TF_NewStatus();
        TFE_Op* op = TFE_NewOp(ctx_1, "ReadVariableOp", status);
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
        TFE_OpAddInput(op, var_handle, status);
        ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
        TFE_Execute(op, &handle_1, &num_retvals, status);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  2. tensorflow/c/eager/c_api_remote_test_util.h

    #define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_
    
    // Run a function containing a MatMul op and check its output.
    // If heavy_load_on_streaming_rpc is true, send some rpc requests before the one
    // which creates a remote input, to simulate a scenario that the remote input
    // is not ready when we start running an op or a function.
    void TestRemoteExecuteSilentCopies(bool async, bool remote, bool func,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Dec 11 22:56:03 GMT 2020
    - 1.4K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.h

    //     the additional sanity checks there seem unnecessary;
    typedef struct TFE_Op TFE_Op;
    
    TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
                                            const char* op_or_function_name,
                                            TF_Status* status);
    TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op);
    
    // Returns the op or function name `op` will execute.
    //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
  4. tensorflow/c/eager/gradients.cc

    namespace internal {
    Status Reset(AbstractOperation* op_, const char* op,
                 const char* raw_device_name, ForwardOperation* forward_op_) {
      forward_op_->op_name = op;
      forward_op_->attrs.Reset(op);
      return op_->Reset(op, raw_device_name);
    }
    Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
                    ForwardOperation* forward_op_) {
      TF_RETURN_IF_ERROR(op_->AddInput(input));
    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)
  5. tensorflow/c/eager/c_api_unified_experimental.cc

    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s) {
      tsl::Set_TF_Status_from_Status(
          s, unwrap(op)->Reset(op_type,
                               /*raw_device_name=*/nullptr));
    }
    
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
                                TF_Status* s) {
      TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  6. tensorflow/c/c_test_util.cc

                        TF_Graph* graph, TF_Status* s, const char* name,
                        TF_Operation** op, const string& op_device, bool check) {
      TF_OperationDescription* desc = TF_NewOperation(graph, op_name, name);
      if (!op_device.empty()) {
        TF_SetDevice(desc, op_device.c_str());
      }
      TF_AddInput(desc, {l, 0});
      TF_AddInput(desc, {r, 0});
      *op = TF_FinishOperation(desc, s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  7. tensorflow/c/eager/custom_device_test.cc

      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get());
      TFE_OpSetAttrString(op.get(), "container", "", 0);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 27 23:39:24 GMT 2020
    - 18.4K bytes
    - Viewed (0)
  8. tensorflow/c/eager/unified_api_testutil.h

                                       absl::Span<AbstractTensorHandle* const>,
                                       absl::Span<AbstractTensorHandle*>)>;
    
    // Runs `model` maybe wrapped in a function call op. This can be thought as
    // being equivalent to the following python code.
    //
    // if use_function:
    //   outputs = tf.function(model)(inputs)
    // else:
    //   outputs = model(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)
  9. tensorflow/c/eager/c_api_unified_experimental.h

    // TODO(srbs): Add APIs for specifying attrs etc.
    // `op_type` must outlive `op`.
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s);
    // `op_name` must outlive `op`.
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
                                TF_Status* s);
    // `attr_name` must outlive `op`.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:00 GMT 2021
    - 7K bytes
    - Viewed (0)
  10. tensorflow/c/experimental/gradients/nn_grad.cc

    GradientFunction* ReluRegisterer(const ForwardOperation& op) {
      return new ReluGradientFunction(op.outputs);
    }
    
    GradientFunction* SparseSoftmaxCrossEntropyWithLogitsRegisterer(
        const ForwardOperation& op) {
      return new SparseSoftmaxCrossEntropyWithLogitsGradientFunction(op.outputs);
    }
    
    GradientFunction* BiasAddRegisterer(const ForwardOperation& op) {
      return new BiasAddGradientFunction(op.attrs);
    }
    
    }  // namespace gradients
    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)
Back to top