Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dsaved_model_to_tfl_flatbuffer.cc56 mlir::func::FuncOp entry_function = nullptr; in HandleInputOutputArraysWithModule() local
62 if (entry_function != nullptr) { in HandleInputOutputArraysWithModule()
66 entry_function = func; in HandleInputOutputArraysWithModule()
69 if (entry_function == nullptr) { in HandleInputOutputArraysWithModule()
75 entry_function->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); in HandleInputOutputArraysWithModule()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
H A Dsignature_with_multiple_entry_points.mlir7 // CHECK-SAME: tf.entry_function = {inputs = "input1:0,input2:0", outputs = "result:0"}
14 // CHECK-SAME: tf.entry_function = {inputs = "input2:0,input1:0", outputs = "result:0"}
18 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
24 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
H A 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"}} {
H A Dsignature.mlir7 // CHECK-SAME: tf.entry_function = {inputs = "serving_default_input2:0,serving_default_input1:0", o…
10 …x5xf32> {tf_saved_model.index_path = ["end_logits"]}) attributes {tf.entry_function = {control_out…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/tests/
H A Dinsert_main_function.mlir25 …or<1xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {inputs = "m…
29 …tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> attributes {tf.entry_function = {inputs = "m…
34 …or<1xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {inputs = "m…
40 …tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> attributes {tf.entry_function = {inputs = "m…
47 …tf_saved_model.index_path = ["PartitionedCall_1:0"]}) attributes {tf.entry_function = {inputs = "m…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtf_saved_model_initialize_variables_in_session_init.mlir8 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
29 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
55 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
69 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
90 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
H A 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…
H A 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…
H A 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.func @main() attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = ""}…
H A Doptimize-arg-operand-constraint.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "output_node"}} {
H A Dtf_saved_model_initialize_variables_in_session_init_fail.mlir7 …attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = "dense_2/Add:0"}, tf…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
H A Dlegalize-tf-variables.mlir7 …1x10xi64> {tf_saved_model.index_path = ["output_1"]}) attributes {tf.entry_function = {control_out…
53 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
74 …(tensor<1x10xui64> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {co…
88 …() attributes {tf.entry_function = {control_outputs = "", inputs = "", outputs = ""}, tf_saved_mod…
104 …(tensor<1x10xf32> {tf_saved_model.index_path = ["output_0"]}) attributes {tf.entry_function = {con…
H A Dpost-quantize-dynamic-range.mlir7 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
22 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
42 …CustomOp(%arg0: tensor<1x1x1x1xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
70 …32>) -> (tensor<*xf32>, tensor<*xf32>, tensor<*xf32>) attributes {tf.entry_function = {inputs = "i…
H A 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"}, …
H A Dtfl_while_outline.mlir11 attributes {tf.entry_function = {outputs = "result"}} {
43 func.func @while2(%cst : tensor<i32>) -> tensor<1xf32> attributes {tf.entry_function = {outputs = "…
86 …> {tf.device = "/device:CPU:0"}) -> tensor<4x?x2xf32> attributes {tf.entry_function = {inputs = "P…
173 attributes {tf.entry_function = {outputs = "result"}} {
H A 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"}} {
H A Dprepare-quantize-post-training.mlir41 …ithoutNorm(%arg0: tensor<1x1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
120 func.func @QuantizeLstmCifg(%arg0: tensor<1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function …
202 …alLstmFull(%arg0: tensor<1x1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
293 …angedInput(%arg0: tensor<1x1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
348 …tionalLstm(%arg0: tensor<1x1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function = {inputs = "i…
402 func.func @QuantizeLstmFull(%arg0: tensor<1x5xf32>) -> tensor<*xf32> attributes {tf.entry_function …
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dtf_to_corert_pipeline_cpurt.mlir25 attributes { tf.entry_function = {control_outputs = "",
61 attributes { tf.entry_function = {control_outputs = "",
102 attributes { tf.entry_function = {control_outputs = "",
142 attributes { tf.entry_function = {control_outputs = "",
173 attributes { tf.entry_function = {control_outputs = "",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/
H A Dtest.mlir10 tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "result:0"},
21 tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "result:0"},
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc1020 auto entry_function = mlir::cast<mlir::func::FuncOp>( in CompileModuleToLlvmIrImpl() local
1024 entry_function, &results->allocations, &results->output_info, in CompileModuleToLlvmIrImpl()
1049 TF_RETURN_IF_ERROR(ir_emitter->EmitLmhloRegion(&entry_function.getBody())); in CompileModuleToLlvmIrImpl()
1077 LowerToJitRt(*mlir_module, entry_function.getName(), in CompileModuleToLlvmIrImpl()
1616 mlir::func::FuncOp entry_function = in CompileLmhloToExecutable() local
1624 TF_RETURN_IF_ERROR(GetMlirAllocationInfo(entry_function, &allocations, in CompileLmhloToExecutable()
1634 TF_RETURN_IF_ERROR(ir_emitter->EmitLmhloRegion(&entry_function.getBody())); in CompileLmhloToExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/
H A Dsignature_def_with_multiple_entry_points.mlir162 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
168 …nsor<?xf32> {tf_saved_model.index_path = ["result"]}) attributes {tf.entry_function = {control_out…
H A Dmetadata.mlir7 attributes {tf.entry_function = {inputs = "input", outputs = "SameNameAsOutput"}} {
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcodecache.py2522 entry_function = "kernel" variable in CppPythonBindingsCodeCache
2646 cls.entry_function,
2648 cls.entry_function,
2652 cls.entry_function,
2653 cls.entry_function,
2654 cls.entry_function,
2655 cls.entry_function,
2667 return getattr(result, cls.entry_function)
2684 entry_function = "inductor_entry_cpp" variable in CppWrapperCodeCache
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
H A Dexport_main_to_flib.mlir4 func.func @main() attributes {tf.entry_function = {inputs = "", outputs = ""}} {
H A Daliasing_arg_attr.mlir7 attributes {tf.entry_function = {inputs = "args_0,args_1", outputs = "rets_0,rets_1"}} {

1234