Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 38 for op (0.12 sec)

  1. tensorflow/c/eager/c_api_test.cc

      TFE_DeleteContextOptions(opts);
    
      TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status);
      TFE_OpSetAttrType(var_op, "dtype", TF_INT64);
      TFE_OpSetAttrShape(var_op, "shape", {}, 0, status);
      const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op);
    
      TFE_Op* copy_op = TFE_NewOp(ctx, "VarHandleOp", status);
      TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT);
      TFE_OpAddAttrs(copy_op, attributes);
      unsigned char is_list = 0;
    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)
  2. tensorflow/c/eager/c_api_unified_experimental_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      // Build an abstract operation.
      auto* add_op = TF_NewAbstractOp(graph_ctx);
      TF_AbstractOpSetOpType(add_op, "Add", status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      TF_AbstractOpSetOpName(add_op, "my_add", status.get());
      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
    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)
  3. tensorflow/c/c_api_test.cc

        TF_DeleteStatus(status);
        ASSERT_EQ(TF_OK, code) << status_msg;
    
        // Test op list.
        TF_Buffer op_list_buf = TF_GetOpList(lib);
        tensorflow::OpList op_list;
        EXPECT_TRUE(op_list.ParseFromArray(op_list_buf.data, op_list_buf.length));
        ASSERT_EQ(op_list.op_size(), 1);
        EXPECT_EQ("TestCApi1", op_list.op(0).name());
        TF_DeleteLibraryHandle(lib);
      }
    #endif  // !defined(TENSORFLOW_NO_SHARED_OBJECTS)
    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)
  4. 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)
  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/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)
  7. 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)
  8. 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)
  9. tensorflow/c/c_api.cc

      delete lib_handle;
    }
    
    TF_Buffer* TF_GetAllOpList() {
      std::vector<tensorflow::OpDef> op_defs;
      tensorflow::OpRegistry::Global()->GetRegisteredOps(&op_defs);
      tensorflow::OpList op_list;
      for (const auto& op : op_defs) {
        *(op_list.add_op()) = op;
      }
      TF_Buffer* ret = TF_NewBuffer();
      TF_CHECK_OK(MessageToBuffer(op_list, ret));
      return ret;
    }
    
    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)
  10. tensorflow/c/eager/c_api_test_util.cc

      TFE_OpAddInput(op, in, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TF_DeleteStatus(status);
    
      TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(in));
      TFE_OpSetAttrInt(op, "group_size", group_size);
      TFE_OpSetAttrInt(op, "group_key", 123);
      TFE_OpSetAttrInt(op, "instance_key", 456);
      TFE_OpSetAttrString(op, "merge_op", "Add", 3);
      TFE_OpSetAttrString(op, "final_op", "Id", 2);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 21 22:37:46 GMT 2024
    - 23.5K bytes
    - Viewed (2)
Back to top