Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_OperationDescription (0.16 sec)

  1. tensorflow/c/c_api.cc

      return new TF_OperationDescription(graph, op_type, oper_name);
    }
    
    TF_OperationDescription* TF_NewOperation(TF_Graph* graph, const char* op_type,
                                             const char* oper_name) {
      mutex_lock l(graph->mu);
      return TF_NewOperationLocked(graph, op_type, oper_name);
    }
    
    void TF_SetDevice(TF_OperationDescription* desc, const char* device) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      static bool classof(const AbstractTensorHandle* ptr) {
        return ptr->getKind() == kGraph;
      }
    
     private:
      TF_Graph* graph_;  // For shape inference.
    };
    
    // GraphOperation wraps and populates a TF_OperationDescription.
    class GraphOperation : public TracingOperation {
     public:
      explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {}
      void Release() override { delete this; }
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

        TF_AddInputList(desc_, inputs, TF_ARRAYSIZE(inputs));
      }
    
      ~CApiColocationTest() override {
        TF_DeleteGraph(graph_);
        TF_DeleteStatus(s_);
      }
    
      void SetViaStringList(TF_OperationDescription* desc,
                            const std::vector<string>& list) {
        std::unique_ptr<const void*[]> list_ptrs;
        std::unique_ptr<size_t[]> list_lens;
        StringVectorToArrays(list, &list_ptrs, &list_lens);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  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 =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Tue Aug 06 23:56:17 UTC 2024
    - 29.4K bytes
    - Viewed (0)
Back to top