Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for set_var_y_data (0.39 sec)

  1. tensorflow/compiler/aot/tests/tfcompile_test.cc

      // x[n+1] = x[n] - 0.1*(x[n-1] + 1.0)
      VariableSequentialUpdatesComp fn(
          XlaCompiledCpuFunction::AllocMode::RESULTS_PROFILES_AND_TEMPS_ONLY);
      float x = 2;
      float y = 1;
      fn.set_var_x_data(&x);
      fn.set_var_y_data(&y);
    
      fn.set_thread_pool(&device);
      // First calculate x[3]
      fn.Run();
      EXPECT_NEAR(x, 1.187f, 1e-6);
    
      // Then calculate x[6]
      fn.Run();
      EXPECT_NEAR(x, 0.594322f, 1e-6);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 26.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/aot/codegen_test_h.golden

      // constness of the buffer passed to `set_var_X_data` but the underlying
      // buffer is not const (and thus the const can be safely const-cast'ed away)
      // unless `set_var_X_data` is called with a pointer to constant storage.
    
      void set_var_myvar_readonly_data(const float* data) {
        set_arg_data(2, data);
      }
      const float* var_myvar_readonly_data() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 01:20:01 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/aot/codegen.cc

      // constness of the buffer passed to `set_var_X_data` but the underlying
      // buffer is not const (and thus the const can be safely const-cast'ed away)
      // unless `set_var_X_data` is called with a pointer to constant storage.
    {{METHODS_VARIABLE}}
    
     private:
      // Number of buffers for the compiled computation.
      static constexpr size_t kNumBuffers = {{NUM_BUFFERS}};
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 01:20:01 UTC 2024
    - 36.8K bytes
    - Viewed (0)
Back to top