Home
last modified time | relevance | path

Searched refs:referenced_func (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc126 FuncOp referenced_func = entry_module_table.lookup<FuncOp>( in EncapsulateFuncAndSerialize() local
130 if (!referenced_func) continue; in EncapsulateFuncAndSerialize()
132 referenced.emplace_back(referenced_func); in EncapsulateFuncAndSerialize()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_rewrite.mlir591 // CHECK-SAME: func @referenced_func
604 %0 = "tf.B"(%arg0) {body = @referenced_func} : (tensor<?xi32>) -> tensor<?xi32>
608 func @referenced_func(%arg0: tensor<?xi32>) -> tensor<?xi32> {
683 // CHECK-COUNT-2: call @referenced_func
684 // CHECK-COUNT-1: func @referenced_func
698 %1 = call @referenced_func(%0) : (tensor<?xi32>) -> tensor<?xi32>
699 %2 = call @referenced_func(%1) : (tensor<?xi32>) -> tensor<?xi32>
703 func @referenced_func(%arg0: tensor<?xi32>) -> tensor<?xi32> {