Home
last modified time | relevance | path

Searched refs:decomposed_callee (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc149 FuncOp decomposed_callee; member
308 call.getLoc(), info.decomposed_callee.getType().getResults(), in HandlePartitionedCallOp()
312 const_cast<FuncOp&>(info.decomposed_callee).getName())); in HandlePartitionedCallOp()
319 int64_t aliased_input = FindAliasedInput(info.decomposed_callee, i); in HandlePartitionedCallOp()
354 info.decomposed_callee = lowered_callee; in HandlePartitionedCallOp()
Dtensor_array_ops_decomposition.cc517 FuncOp decomposed_callee; member
753 call.getLoc(), info.decomposed_callee.getType().getResults(), in HandlePartitionedCallOp()
757 const_cast<FuncOp&>(info.decomposed_callee).getName())); in HandlePartitionedCallOp()
812 info.decomposed_callee = lowered_callee; in HandlePartitionedCallOp()
Dtensor_list_ops_decomposition.cc114 FuncOp decomposed_callee; member
458 call.getLoc(), info.decomposed_callee.getType().getResults(), in HandlePartitionedCallOp()
462 const_cast<FuncOp&>(info.decomposed_callee).getName())); in HandlePartitionedCallOp()
504 info.decomposed_callee = lowered_callee; in HandlePartitionedCallOp()