Home
last modified time | relevance | path

Searched defs:control_outputs (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_flags.h46 std::vector<string> control_outputs; member
Dimport_model.cc2253 auto control_outputs = in Convert() local
2288 auto control_outputs = in Convert() local
2487 llvm::ArrayRef<std::string> control_outputs, in GetControlRetsFromGraph()
3623 const std::vector<std::string> control_outputs) { in ConvertGraph()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc44 std::vector<ControlOutput> control_outputs, const int graph_def_version, in GrapplerFunctionItem()
102 const std::vector<ControlOutput>& GrapplerFunctionItem::control_outputs() in control_outputs() function in tensorflow::grappler::GrapplerFunctionItem
282 std::vector<ControlOutput> control_outputs; in MakeGrapplerFunctionItem() local
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc318 std::vector<const Node*> control_outputs; in GraphToFunctionDefHelper() local
375 const std::vector<const Node*>& control_outputs, in GraphToFunctionDef()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc291 absl::flat_hash_set<Node*> control_outputs; in BuildXlaLaunchOps() local
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc558 std::vector<TF_Operation*> control_outputs( in TF_OperationGetControlOutputs_wrapper() local
583 const std::vector<TF_Operation*>* control_outputs, in TF_GraphToFunction_wrapper()
Dtf_session_wrapper.cc198 const char* description) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc155 int ncontrol_outputs, const TF_Operation* const* control_outputs, in TF_GraphToFunctionWithControlOutputs()
Dc_api.cc1217 TF_Operation** control_outputs, in TF_OperationGetControlOutputs()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc534 std::vector<gtl::InlinedVector<std::pair<int, int>, 2>> control_outputs( in TransitiveReduction() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc2772 gtl::FlatSet<Node*> control_outputs; in BuildTPUReplicateOps() local