Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for TF_FinishOperation (0.34 sec)

  1. tensorflow/c/c_test_util.cc

      TF_AddInputList(desc, inputs, 2);
      return TF_FinishOperation(desc, s);
    }
    
    void NegHelper(TF_Operation* n, TF_Graph* graph, TF_Status* s, const char* name,
                   TF_Operation** op) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "Neg", name);
      TF_Output neg_input = {n, 0};
      TF_AddInput(desc, neg_input);
      *op = TF_FinishOperation(desc, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  2. tensorflow/c/c_api_test.cc

      TF_SetAttrIntList(desc, "v", nullptr, 0);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", 0, TF_ATTR_INT, -1);
    }
    
    TEST_F(CApiAttributesTest, Names) {
      auto desc = init("string");
      TF_SetAttrString(desc, "v", "bunny", 5);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  3. tensorflow/c/c_api.h

        TF_Graph* graph, const char* op_type, const char* oper_name);
    
    // Operation will only be added to *graph when TF_FinishOperation() is
    // called (assuming TF_FinishOperation() does not return an error).
    // *graph must not be deleted until after TF_FinishOperation() is
    // called.
    TF_CAPI_EXPORT extern TF_OperationDescription* TF_NewOperation(
        TF_Graph* graph, const char* op_type, const char* oper_name);
    
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  4. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TF_OperationDescription* placeholder_desc =
          TF_NewOperation(body.get(), "Placeholder", "Placeholder");
      TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT);
      TF_Operation* placeholder_op = TF_FinishOperation(placeholder_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
      TF_Output x{placeholder_op, 0};
    
      TF_OperationDescription* reduce_desc =
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  5. tensorflow/c/eager/c_api_experimental_test.cc

      TF_Status* status = TF_NewStatus();
      TF_Operation* arg = TF_FinishOperation(arg_descr, status);
      ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
      TF_OperationDescription* id_descr =
          TF_NewOperation(function_graph, "Identity", "id");
      TF_SetAttrType(id_descr, "T", TF_INT32);
      TF_AddInput(id_descr, {arg, 0});
      TF_Operation* id = TF_FinishOperation(id_descr, status);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        auto* tf_opdesc = op_.release();
        if (tf_opdesc == nullptr) {
          return errors::InvalidArgument("AbstractOp is incomplete.");
        }
        TF_Status* s = TF_NewStatus();
        auto* operation = TF_FinishOperation(tf_opdesc, s);
        TF_RETURN_IF_ERROR(StatusFromTF_Status(s));
        TF_DeleteStatus(s);
        *num_retvals = TF_OperationNumOutputs(operation);
        for (int i = 0; i < *num_retvals; ++i) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. tensorflow/c/c_api_function_test.cc

          TF_AddInput(desc, input);
        }
        // Set device to CPU because some ops inside the function might not be
        // available on GPU.
        TF_SetDevice(desc, "/cpu:0");
        *op = TF_FinishOperation(desc, s_);
        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        ASSERT_NE(*op, nullptr);
      }
    
      FunctionDef fdef() {
        tensorflow::FunctionDef fdef;
        EXPECT_TRUE(GetFunctionDef(func_, &fdef));
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  8. tensorflow/c/c_api.cc

        } else if (ret != nullptr) {
          desc->graph->graph.RemoveNode(ret);
          ret = nullptr;
        }
      }
    
      delete desc;
    
      return ToOperation(ret);
    }
    
    TF_Operation* TF_FinishOperation(TF_OperationDescription* desc,
                                     TF_Status* status) {
      mutex_lock l(desc->graph->mu);
      return TF_FinishOperationLocked(desc, status);
    }
    
    // TF_Operation functions
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top