Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc163 decomposed_partitioned_call_callees) { in HandleWhileOp()
184 decomposed_partitioned_call_callees))) { in HandleWhileOp()
192 decomposed_partitioned_call_callees))) { in HandleWhileOp()
231 decomposed_partitioned_call_callees) { in HandleIfOp()
246 decomposed_partitioned_call_callees)) || in HandleIfOp()
248 decomposed_partitioned_call_callees))) { in HandleIfOp()
288 decomposed_partitioned_call_callees) { in HandlePartitionedCallOp()
289 auto emplace_res = decomposed_partitioned_call_callees->try_emplace( in HandlePartitionedCallOp()
369 decomposed_partitioned_call_callees))) { in HandlePartitionedCallOp()
470 decomposed_partitioned_call_callees) { in HandleRegionControlFlowOps()
[all …]
Dtensor_list_ops_decomposition.cc167 decomposed_partitioned_call_callees) { in HandleWhileOp()
184 decomposed_partitioned_call_callees))) { in HandleWhileOp()
196 decomposed_partitioned_call_callees))) { in HandleWhileOp()
227 decomposed_partitioned_call_callees) { in HandleCaseOrIfOp()
249 decomposed_partitioned_call_callees))) in HandleCaseOrIfOp()
285 decomposed_partitioned_call_callees) { in HandleWhileRegionOp()
306 decomposed_partitioned_call_callees))) { in HandleWhileRegionOp()
317 decomposed_partitioned_call_callees))) { in HandleWhileRegionOp()
349 decomposed_partitioned_call_callees) { in HandleIfRegionOp()
355 decomposed_partitioned_call_callees))) in HandleIfRegionOp()
[all …]
Dtensor_array_ops_decomposition.cc559 decomposed_partitioned_call_callees) { in HandleWhileOp() argument
580 decomposed_partitioned_call_callees)) || in HandleWhileOp()
582 decomposed_partitioned_call_callees))) { in HandleWhileOp()
643 decomposed_partitioned_call_callees) { in HandleIfOp() argument
664 decomposed_partitioned_call_callees)) || in HandleIfOp()
666 decomposed_partitioned_call_callees))) { in HandleIfOp()
726 decomposed_partitioned_call_callees) { in HandlePartitionedCallOp()
727 auto emplace_res = decomposed_partitioned_call_callees->try_emplace( in HandlePartitionedCallOp()
801 decomposed_partitioned_call_callees))) { in HandlePartitionedCallOp()
833 decomposed_partitioned_call_callees) { in HandleRegionControlFlowOps()
[all …]