Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for func (0.14 sec)

  1. tensorflow/c/c_api_function_test.cc

      // Define
      TF_Operation* feed = Placeholder(func_graph_, s_);
      TF_Operation* split = Split3(feed, func_graph_, s_);
      DefineT(-1, {}, {{feed, 0}}, {{split, 0}, {split, 2}}, {});
    
      // Use, run, and verify
      TF_Operation* func_feed = Placeholder(host_graph_, s_);
      TF_Operation* func_op = Use({func_feed});
      RunT({{func_feed, Int32Tensor({1, 2, 3, 4, 5, 6})}},
           {{func_op, 0}, {func_op, 1}}, {{1, 2}, {5, 6}});
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  2. tensorflow/c/eager/c_api_unified_experimental.cc

    void TF_DeleteAbstractFunction(TF_AbstractFunction* func) {
      unwrap(func)->Unref();
    }
    
    void TF_ExecutionContextRegisterFunction(TF_ExecutionContext* ctx,
                                             TF_AbstractFunction* func,
                                             TF_Status* s) {
      tsl::Set_TF_Status_from_Status(s,
                                     unwrap(ctx)->RegisterFunction(unwrap(func)));
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  3. tensorflow/c/eager/graph_function.h

      absl::StatusOr<core::RefCountPtr<FunctionRecord>> GetFunctionRecord()
          override {
        return func_record_.GetNewRef();
      }
    
      // For LLVM style RTTI.
      static bool classof(const AbstractFunction* ptr) {
        return ptr->getKind() == kGraph;
      }
    
     private:
      core::RefCountPtr<FunctionRecord> func_record_;
    };
    }  // namespace graph
    }  // namespace tracing
    }  // namespace tensorflow
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Mar 04 19:49:06 GMT 2024
    - 1.8K bytes
    - Viewed (0)
  4. tensorflow/c/eager/unified_api_testutil.cc

        {
          AbstractContextPtr func_ctx(BuildFunction(fn_name));
          std::vector<AbstractTensorHandle*> func_inputs;
          func_inputs.reserve(inputs.size());
          TF_RETURN_IF_ERROR(
              CreateParamsForInputs(func_ctx.get(), inputs, &func_inputs));
          std::vector<AbstractTensorHandle*> model_outputs;
          model_outputs.resize(outputs.size());
          TF_RETURN_IF_ERROR(model(func_ctx.get(), absl::MakeSpan(func_inputs),
    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)
  5. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      }
      Status SetAttrFunctionName(const char* attr_name, const char* value,
                                 size_t length) override {
        tensorflow::NameAttrList func_name;
        func_name.set_name(string(value, value + length));
        op_->node_builder.Attr(attr_name, func_name);
        return absl::OkStatus();
      }
      Status SetAttrTensor(const char* attr_name,
                           AbstractTensorInterface* tensor) override {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  6. tensorflow/c/eager/c_api_test.cc

    REGISTER_KERNEL_BUILDER(
        Name("TestCommUnavailable").Device(tensorflow::DEVICE_DEFAULT),
        TestUnavailableErrorOp);
    
    string FunctionWithErrorOp(const tensorflow::StringPiece op_name) {
      const std::string& func_str =
          "    signature {"
          "      name: 'FunctionWith__OP_NAME__'"
          "      output_arg {"
          "        name: 'out'"
          "        type: DT_STRING"
          "      }"
          "    }"
    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)
  7. tensorflow/c/eager/c_api.cc

                    TF_Status* status) {
      TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status);
      for (const auto& attr : func.attr()) {
        if (!status->status.ok()) return nullptr;
        SetOpAttrValueScalar(ctx, func_op, attr.second, attr.first.data(), status);
        if (!status->status.ok()) return nullptr;
      }
      return func_op;
    }
    }  // namespace
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  8. tensorflow/c/c_api.cc

    }
    
    void TF_SetAttrFuncName(TF_OperationDescription* desc, const char* attr_name,
                            const char* value, size_t length) {
      tensorflow::NameAttrList func_name;
      func_name.set_name(string(value, value + length));
      desc->node_builder.Attr(attr_name, func_name);
    }
    
    void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name,
                         const int64_t* dims, int num_dims) {
    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)
  9. tensorflow/c/c_api_function.cc

      }
      const auto len = std::min(max_func, static_cast<int>(lib.function_size()));
      for (int i = 0; i < len; ++i) {
        TF_Function* func = new TF_Function();
        func->record = new tensorflow::FunctionRecord(lib.function(i), {}, false);
        funcs[i] = func;
      }
      status->status = absl::OkStatus();
      return len;
    }
    
    void TF_FunctionToFunctionDef(TF_Function* func, TF_Buffer* output_func_def,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  10. tensorflow/c/eager/graph_function.cc

    namespace tracing {
    namespace graph {
    GraphFunction::GraphFunction(FunctionDef fdef)
        : AbstractFunction(kGraph),
          func_record_(new FunctionRecord(std::move(fdef), {}, true)) {}
    GraphFunction::~GraphFunction() {}
    Status GraphFunction::GetFunctionDef(const FunctionDef **fdef) {
      *fdef = &(func_record_->fdef());
      return absl::OkStatus();
    }
    }  // namespace graph
    }  // namespace tracing
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Mar 04 19:49:06 GMT 2024
    - 1.3K bytes
    - Viewed (0)
Back to top