Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc222 llvm::MutableArrayRef<tf_device::LaunchOp> execute_launches, in HandleCompileAndExecutes() argument
233 execute_launches.front()->getParentOfType<tf_device::ReplicateOp>(); in HandleCompileAndExecutes()
236 llvm::zip(execute_launches, input_mappings)) { in HandleCompileAndExecutes()
289 llvm::SmallVector<tf_device::LaunchOp, 4> execute_launches; in runOnFunction() local
290 execute_launches.reserve(compile_launch.getNumResults() - 1); in runOnFunction()
299 execute_launches.push_back(execute_launch); in runOnFunction()
302 HandleCompileAndExecutes(compile_launch, execute_launches, in runOnFunction()
Dtpu_colocate_composite_resource_ops.cc109 llvm::SmallVector<tf_device::LaunchOp, 8> execute_launches; in runOnFunction() local
114 execute_launches.push_back(op); in runOnFunction()
118 for (auto execute_launch : execute_launches) { in runOnFunction()
Dtpu_merge_variables_with_execute.cc599 llvm::SmallVector<tf_device::LaunchOp, 8> execute_launches; in runOnFunction() local
604 execute_launches.push_back(op); in runOnFunction()
607 for (auto execute_launch : execute_launches) { in runOnFunction()