Home
last modified time | relevance | path

Searched defs:control_deps (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dnode_matchers.h100 const std::optional<NodeSeqMatcher>& control_deps() const { in control_deps() function
131 void set_control_deps(NodeSeqMatcher control_deps) { in set_control_deps()
224 impl::NodeMatcherProperties CtrlDeps(Ts... control_deps) { in CtrlDeps()
H A Dincrease_dynamism_for_auto_jit_pass.cc149 const std::vector<const Edge*>& control_deps) in ConstantCache()
174 std::vector<const Edge*> control_deps, Output* size) { in ComputeSliceSize()
H A Dnode_matchers.cc180 std::vector<const Node*> control_deps; in MatchAndExplain() local
465 impl::NodeMatcherProperties impl::CtrlDeps( in CtrlDeps()
H A Dencapsulate_util.cc279 std::vector<string> control_deps; in PostprocessControlEdgesBetweenOutsideCompilations() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dconstant_folding.cc319 gtl::FlatSet<Node*>& control_deps = (*constant_control_deps)[n]; in ConsiderConstantFoldableNode() local
493 const Tensor& constant, const gtl::FlatSet<Node*>& control_deps, in ReplaceTensorWithConstant()
674 const gtl::FlatSet<Node*>& control_deps = in ConstantFold() local
H A Ddirect_session_test.cc1448 auto make_expensive_noop = [&g](gtl::ArraySlice<Node*> control_deps) { in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dscope.cc120 std::vector<Operation> control_deps, bool clear_control_deps) in Impl()
298 const std::vector<Operation>& Scope::control_deps() const { in control_deps() function in tensorflow::Scope
H A Dscope_internal.h48 const std::vector<Operation>& control_deps() const { return control_deps_; } in control_deps() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer.cc243 absl::flat_hash_set<string> control_deps; member
469 absl::flat_hash_set<string>* control_deps) { in PushDownConstInputs()
728 absl::flat_hash_set<string> control_deps; in SpecializeFunction() local
H A Dimplementation_selector.cc174 std::vector<std::string> control_deps; in UpdateNodeDef() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dgraph_info_test.cc114 const std::vector<std::pair<int, int>>& control_deps) { in PartitionGraph() argument
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.cc1870 const std::vector<tensorflow::Operation>& control_deps, in CopyGraph()