- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for set_is_stateful (0.1 sec)
-
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
{"shape_shape:output:0", "minval:output:0", "maxval:output:0"}, {{"Tout", DataType::DT_INT32}, {"T", DataType::DT_INT32}}}}, /*ret_def=*/{{"out", "shape:output:0"}}); func.mutable_signature()->set_is_stateful(true); *flib_def.add_function() = std::move(func); TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); NodeDef call_node; call_node.set_name("fn_call"); call_node.set_op("Stateful_func");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0)