Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetModelOps (0.12 sec)

  1. tensorflow/c/experimental/ops/gen/common/controller.h

     public:
      explicit Controller(PathConfig path_config, Env* env = Env::Default());
      virtual ~Controller();
      const void WriteFile(const string& file_path, const SourceCode& code) const;
      const std::vector<OpSpec>& GetModelOps() const;
    
     private:
      void InitializeOpApi();
      void BuildModel();
    
      // Data model: Ops to generate
      std::vector<OpSpec> operators_;
    
      // Configuration
      Env* env_;
      PathConfig path_config_;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 21 18:51:25 UTC 2021
    - 1.8K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/ops/gen/cpp/cpp_generator.cc

          path_config_(path_config) {}
    
    SourceCode CppGenerator::GenerateOneFile(
        cpp::RendererContext::Mode mode) const {
      SourceCode generated_code;
      const std::vector<OpSpec> ops(controller_.GetModelOps());
      std::vector<cpp::OpView> views(ops.begin(), ops.end());
      cpp::RendererContext context{mode, generated_code, cpp_config_, path_config_};
      cpp::CppFileRenderer(context, views).Render();
      return generated_code;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 21 18:51:25 UTC 2021
    - 2.2K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/ops/gen/common/controller.cc

                                     const SourceCode& code) const {
      TF_CHECK_OK(WriteStringToFile(env_, file_path, code.Render())) << file_path;
    }
    
    const std::vector<OpSpec>& Controller::GetModelOps() const {
      return operators_;
    }
    
    void Controller::InitializeOpApi() {
      OpRegistry::Global()->Export(false, &op_list_);
    
      // Load matching API defs for each Op. Paths are visited in order, allowing
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 09:51:28 UTC 2024
    - 3.2K bytes
    - Viewed (0)
Back to top