Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc518 auto execute_inputs = input_list[core]; in BuildParallelExecuteOp() local
519 execute_inputs.emplace_back(compile_op->getResult(core + 1)); in BuildParallelExecuteOp()
522 result = BuildExecuteOp(core, output_sharding_config, execute_inputs, in BuildParallelExecuteOp()
687 llvm::SmallVector<Value, 4> execute_inputs(cluster_func.getOperands()); in Rewrite() local
688 execute_inputs.emplace_back(compile_op->getResult(1)); in Rewrite()
692 /*core_id=*/0, output_shardings, execute_inputs, cluster_func, builder, in Rewrite()