Searched refs:control_rets (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.h | 147 llvm::ArrayRef<std::string> control_rets, llvm::StringRef device_type,
163 llvm::ArrayRef<std::string> control_rets,
|
| D | compile_mlir_util.cc | 652 llvm::ArrayRef<std::string> control_rets, in GraphToModule() argument
662 config.control_outputs = control_rets; in GraphToModule()
682 llvm::ArrayRef<std::string> control_rets, in BuildHloFromGraph() argument
690 TF_RETURN_IF_ERROR(GraphToModule(graph, control_rets, flib_def, debug_info, in BuildHloFromGraph()
698 llvm::ArrayRef<std::string> control_rets, llvm::StringRef device_type, in CompileGraphToXlaHlo() argument
707 TF_RETURN_IF_ERROR(GraphToModule(graph, control_rets, flib_def, debug_info, in CompileGraphToXlaHlo()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 301 std::vector<std::string> control_rets; in CompileSingleOp() local
303 control_rets.push_back(node_def.name()); in CompileSingleOp()
306 *graph, mlir::SpanToArrayRef<XlaCompiler::Argument>(args), control_rets, in CompileSingleOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | hash_utils.cc | 486 std::vector<const NodeDef*> control_rets; in HashFunction() local
488 control_rets.push_back(&control_ret_node->def()); in HashFunction()
492 HashControlInputs(control_rets, &control_ret_nodes_hash)); in HashFunction()
|