Home
last modified time | relevance | path

Searched refs:entry_function (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dinput_output_names_attr.mlir3 // Tests input and output names from FlatBuffer are added to `tf.entry_function` attribute.
7 // CHECK: attributes {tf.entry_function = {inputs = "input0,input1", outputs = "output0,output1"}}
8 attributes {tf.entry_function = {inputs = "input0,input1", outputs = "output0,output1"}} {
Dpruning_function_input_as_output.mlir4 func @main(%arg0: tensor<4xf32>) -> tensor<4xf32> attributes {tf.entry_function = {inputs = "mul"}}…
Dunranked_function_output.mlir5 // CHECK: func @main(%arg0: tensor<1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs…
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc51 mlir::FuncOp entry_function = nullptr; in HandleInputOutputArraysWithModule() local
57 if (entry_function != nullptr) { in HandleInputOutputArraysWithModule()
61 entry_function = func; in HandleInputOutputArraysWithModule()
64 if (entry_function == nullptr) { in HandleInputOutputArraysWithModule()
70 entry_function->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); in HandleInputOutputArraysWithModule()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_saved_model_lift_variables.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
32 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
42 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
70 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Dtf_saved_model_lift_variables_invalid_session.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
25 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
Doptimize-arg-operand-constraint.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "output_node"}} {
Dgraph_pruning.mlir158 // "tf.entry_function" attribute defined is ignored by the pruning pass: this
175 // "tf.entry_function" attribute defined with no feed/fetch/target nodes is
178 func @main() attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = ""}} {
Dtf_data_fuse_pmap_and_batch.mlir5 func @fuse_pmap_and_batch() -> tensor<!tf.variant> attributes {tf.entry_function = {control_outputs…
Dtf_data_fuse_map_and_batch.mlir5 func @fuse_map_and_batch() -> tensor<!tf.variant> attributes {tf.entry_function = {control_outputs …
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dinsert_call_once_op.mlir19 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "hash_table_L…
35 …attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "output:0"}, …
Dinitialize_variables.mlir7 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
48 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
Dlegalize-tf-variables.mlir7 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
42 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
Doptimize_functional_ops.mlir69 attributes {tf.entry_function = {inputs = "input", outputs = "Conv2D"}} {
102 attributes {tf.entry_function = {inputs = "input", outputs = "Conv2D"}} {
138 attributes {tf.entry_function = {inputs = "input", outputs = "Conv2D"}} {
Dtfl_while_outline.mlir11 attributes {tf.entry_function = {outputs = "result"}} {
43 func @while2(%cst : tensor<i32>) -> tensor<1xf32> attributes {tf.entry_function = {outputs = "resul…
86 …> {tf.device = "/device:CPU:0"}) -> tensor<4x?x2xf32> attributes {tf.entry_function = {inputs = "P…
173 attributes {tf.entry_function = {outputs = "result"}} {
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
Dmetadata.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "SameNameAsOutput"}} {
Dtf_entry_function.mlir4 func @serving_default(%arg0: tensor<3x2xf32>) -> tensor<3x2xf32> attributes {tf.entry_function = {i…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Doutput-shapes-attr.mlir4 attributes {tf.entry_function = {inputs = "input0", outputs = "output0"}} {
Dtf_add.mlir4 attributes {tf.entry_function = {inputs = "input0,input1", outputs = "Add"}} {
Dpreserve-entry-func-names.mlir4 attributes {tf.entry_function = {inputs = "foo,bar", outputs = "Add"}} {
Ddevice-arg-retval-attr.mlir8 attributes {tf.entry_function = {inputs = "args_0,args_1", outputs = "rets_0,rets_1"}} {
Dfunction-resource-args.mlir3 func @main() -> tensor<*x!tf.resource> attributes {tf.entry_function = {inputs = "", outputs = "fun…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/end2end/
Dcustom_opdef.pbtxt40 # CHECK: attributes {tf.entry_function = {control_outputs = "", inputs = "input0,input1", outputs =…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2exec/
Dtfl_while_op.mlir23 attributes {tf.entry_function = {outputs = "result,pconst"}} {
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Darg-retval-attrs.pbtxt155 # CHECK-SAME: attributes {tf.entry_function = {control_outputs = "", inputs = "arg0,arg1,arg2", out…

12