Home
last modified time | relevance | path

Searched refs:session_initializer (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_remove_vars_in_session_initializer.mlir11 …// expected-error@+1 {{'tf_saved_model.session_initializer' op the initializer function does not e…
12 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
19 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
35 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
51 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
72 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
Dtf_saved_model_ops_invalid.mlir280 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
288 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
299 "tf_saved_model.session_initializer"() { initializer = @init } : () -> ()
300 // expected-error@+1 {{there must be no more than one session_initializer op}}
301 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
339 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
347 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
359 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
360 // expected-error@+1 {{there must be no more than one session_initializer op}}
361 "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
[all …]
Dtf_saved_model_ops.mlir5 // CHECK: tf_saved_model.session_initializer
6 "tf_saved_model.session_initializer"() {
Dtf_saved_model_optimize_global_tensors.mlir238 "tf_saved_model.session_initializer"() {initializers = [@legacy_init_op]} : () -> ()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc80 static LogicalResult Verify(SessionInitializerOp session_initializer) { in Verify() argument
82 session_initializer->getParentOfType<ModuleOp>()); in Verify()
84 for (auto sym_ref : session_initializer.initializers()) { in Verify()
89 return session_initializer.emitOpError() in Verify()
93 return session_initializer.emitOpError() in Verify()
99 return session_initializer.emitOpError() in Verify()
103 return session_initializer.emitOpError() in Verify()
Dtf_saved_model_ops.td132 def TfSavedModel_SessionInitializerOp: TfSavedModel_Op<"session_initializer"> {
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dinitialize_variables.mlir21 …// CHECK: "tf_saved_model.session_initializer"() {initializers = [@SessionInitializerFunction]} : …
43 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
44 // CHECK: "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
Dlegalize-tf-variables.mlir37 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
38 // CHECK: "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()
Dinsert_call_once_op.mlir6 "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> ()