| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers.h | 100 const std::optional<NodeSeqMatcher>& control_deps() const { in control_deps() function
131 void set_control_deps(NodeSeqMatcher control_deps) { in set_control_deps() argument
133 control_deps_ = std::move(control_deps); in set_control_deps()
143 !control_deps().has_value() && !attr().has_value(); in IsEmpty()
163 absl::Span<const ::testing::Matcher<const Node*>> control_deps);
224 impl::NodeMatcherProperties CtrlDeps(Ts... control_deps) { in CtrlDeps() argument
225 return impl::CtrlDeps({control_deps...}); in CtrlDeps()
|
| D | node_matchers.cc | 180 std::vector<const Node*> control_deps; in MatchAndExplain() local
183 control_deps.push_back(e->src()); in MatchAndExplain()
189 !control_dep_set->MatchAndExplain(control_deps, &inner_listener)) { in MatchAndExplain()
420 if (prop.control_deps()) { in NodeWith()
423 ::testing::UnorderedElementsAreArray(*prop.control_deps()); in NodeWith()
466 absl::Span<const ::testing::Matcher<const Node*>> control_deps) { in CtrlDeps()
468 absl::c_copy(control_deps, std::back_inserter(control_deps_vector)); in CtrlDeps()
|
| D | increase_dynamism_for_auto_jit_pass.cc | 149 const std::vector<const Edge*>& control_deps) in ConstantCache() argument
150 : scope_(s), control_deps_(control_deps) {} in ConstantCache()
174 std::vector<const Edge*> control_deps, Output* size) { in ComputeSliceSize() argument
194 ConstantCache constant_pool(host_scope, control_deps); in ComputeSliceSize()
224 for (const Edge* e : control_deps) { in ComputeSliceSize()
|
| D | encapsulate_util.cc | 279 std::vector<string> control_deps; in PostprocessControlEdgesBetweenOutsideCompilations() local
282 &control_deps); in PostprocessControlEdgesBetweenOutsideCompilations()
291 for (const string& control_input : control_deps) { in PostprocessControlEdgesBetweenOutsideCompilations()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 319 gtl::FlatSet<Node*>& control_deps = (*constant_control_deps)[n]; in ConsiderConstantFoldableNode() local
330 control_deps.insert(e->src()); in ConsiderConstantFoldableNode()
337 control_deps.insert(parent_deps.begin(), parent_deps.end()); in ConsiderConstantFoldableNode()
493 const Tensor& constant, const gtl::FlatSet<Node*>& control_deps, in ReplaceTensorWithConstant() argument
572 if (control_deps.empty()) { in ReplaceTensorWithConstant()
575 for (Node* node : control_deps) { in ReplaceTensorWithConstant()
674 const gtl::FlatSet<Node*>& control_deps = in ConstantFold() local
678 control_deps, opts.max_constant_size_in_bytes, generate_new_name)) { in ConstantFold()
|
| D | direct_session_test.cc | 1448 auto make_expensive_noop = [&g](gtl::ArraySlice<Node*> control_deps) { in TEST() argument
1451 for (Node* control_dep : control_deps) { in TEST()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.cc | 120 std::vector<Operation> control_deps, bool clear_control_deps) in Impl() argument
129 : (control_deps.insert(control_deps.begin(), in Impl()
132 control_deps)), in Impl()
298 const std::vector<Operation>& Scope::control_deps() const { in control_deps() function in tensorflow::Scope
430 const gtl::ArraySlice<Operation>& control_deps) const { in WithControlDependencies()
433 std::vector<Operation>(control_deps.begin(), control_deps.end()), in WithControlDependencies()
|
| D | scope.h | 129 const gtl::ArraySlice<Operation>& control_deps) const;
232 const std::vector<Operation>& control_deps() const;
|
| D | scope_internal.h | 48 const std::vector<Operation>& control_deps() const { return control_deps_; } in control_deps() function
74 std::vector<Operation> control_deps, bool clear_control_deps);
|
| D | scope_test.cc | 145 EXPECT_EQ(c.control_deps().size(), 2); in TEST()
147 EXPECT_EQ(c_c.control_deps().size(), 3); in TEST()
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla.py | 371 output_tensors, control_deps = _postprocess_flat_outputs(outputs)
373 output_tensors, control_deps = _postprocess_non_flat_outputs(outputs)
385 return control_flow_ops.group(control_deps, name='output_0')
392 with ops.control_dependencies(control_deps):
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | implementation_selector.cc | 174 std::vector<std::string> control_deps; in UpdateNodeDef() local
177 control_deps.push_back(node_def->input(i)); in UpdateNodeDef()
218 for (std::string& control : control_deps) in UpdateNodeDef()
|
| D | function_optimizer.cc | 243 absl::flat_hash_set<string> control_deps; member
469 absl::flat_hash_set<string>* control_deps) { in PushDownConstInputs() argument
475 control_deps->insert(input); in PushDownConstInputs()
517 if (!specialization.control_deps.empty()) { in RemovePushedDownConstInputs()
524 for (const string& ctrl : specialization.control_deps) { in RemovePushedDownConstInputs()
728 absl::flat_hash_set<string> control_deps; in SpecializeFunction() local
730 &control_deps)); in SpecializeFunction()
772 control_deps, signature.active_outputs, output_mapping}; in SpecializeFunction()
|
| /external/tensorflow/tensorflow/lite/ |
| D | graph_info_test.cc | 114 const std::vector<std::pair<int, int>>& control_deps) { in PartitionGraph() argument
118 &graph, nodes_to_partition_int_array, control_deps, subgraphs), in PartitionGraph()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 1532 output_tensors, control_deps, pack_template = _postprocess_flat_outputs(
1535 output_tensors, control_deps, pack_template = (
1553 output_tensors, control_deps,
1582 control_flow_ops.group(control_deps, name="shard_%d" % i)
1606 with ops.control_dependencies(control_deps):
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 1870 const std::vector<tensorflow::Operation>& control_deps, in CopyGraph() argument
1886 for (const tensorflow::Operation& op : control_deps) { in CopyGraph()
2031 scope.impl()->control_deps(), ¶ms->cond_output, in TF_FinishWhileHelper()
2047 scope.impl()->name(), scope.impl()->control_deps(), in TF_FinishWhileHelper()
|