Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dparallel_execute_to_islands.cc94 tf_device::ParallelExecuteOp parallel_execute_op, OpBuilder* builder, in ExpandParallelExecuteToIslands()
124 tf_device::ParallelExecuteOp parallel_execute_op) { in CreateIslandsFromParallelExecute()
197 auto parallel_execute_op = in runOnOperation() local
H A Dtpu_merge_variables_with_execute.cc365 Operation* parallel_execute_op = parallel_execute.getOperation(); in ReplaceParallelExecute() local
H A Dtpu_extract_outside_compilation.cc728 auto parallel_execute_op = builder.create<tf_device::ParallelExecuteOp>( in CreateParallelExecuteForOutsideCompilation() local