- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for set_arg_data (0.2 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/codegen.cc
TF_RETURN_IF_ERROR( AddRewritesForShape(i, xla::Shape(ps.parameters(i)), &rewrites)); const string code = R"( void set_arg{{NAME}}_data(const void* data) { set_arg_data({{I}}, data); } {{TYPE}}* arg{{NAME}}_data() { return static_cast<{{TYPE}}*>(arg_data({{I}})); } {{TYPE}}& arg{{NAME}}({{DIM_VARS}}) { return (*static_cast<{{TYPE}}(*){{DIM_SIZES}}>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K 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)