Home
last modified time | relevance | path

Searched defs:parallel_execute_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dparallel_execute_to_islands.cc101 tf_device::ParallelExecuteOp parallel_execute_op, OpBuilder* builder, in ExpandParallelExecuteToIslands()
131 tf_device::ParallelExecuteOp parallel_execute_op) { in CreateIslandsFromParallelExecute()
204 auto parallel_execute_op = in runOnFunction() local
Dtpu_merge_variables_with_execute.cc365 Operation* parallel_execute_op = parallel_execute.getOperation(); in ReplaceParallelExecute() local
Dtpu_rewrite_pass.cc458 OpBuilder* builder, tf_device::ParallelExecuteOp* parallel_execute_op) { in BuildParallelExecuteOp()
Dtpu_extract_outside_compilation.cc729 auto parallel_execute_op = builder.create<tf_device::ParallelExecuteOp>( in CreateParallelExecuteForOutsideCompilation() local