- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for set_arg0_data (0.34 sec)
-
tensorflow/compiler/aot/codegen_test_h.golden
// Returns a reference to the value of type T for positional argument N, // with dim indices specifying which value. No bounds checking is performed // on dim indices. void set_arg0_data(const void* data) { set_arg_data(0, data); } float* arg0_data() { return static_cast<float*>(arg_data(0)); } float& arg0(size_t dim0, size_t dim1) { return (*static_cast<float(*)[1][2]>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/aot/tests/tfcompile_test.cc
} // Run tests that use set_argN_data separately, to avoid accidentally re-using // non-existent buffers. TEST(TFCompileTest, Add_SetArg) { AddComp add( XlaCompiledCpuFunction::AllocMode::RESULTS_PROFILES_AND_TEMPS_ONLY); int32 arg_x = 10; int32 arg_y = 32; add.set_arg0_data(&arg_x); add.set_arg1_data(&arg_y); EXPECT_EQ(add.arg0_data(), add.arg_data(0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 26.4K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen.cc
// Arg methods for managing input buffers. Buffers are in row-major order. // There is a set of methods for each positional argument, with the following // general form: // // void set_argN_data(void* data) // Sets the buffer of type T for positional argument N. May be called in // any AllocMode. Must be called before Run to have an effect. Must be
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0)