/external/tensorflow/tensorflow/compiler/jit/ |
node_matchers.h | 100 const absl::optional<NodeSeqMatcher>& control_deps() const { function in class:tensorflow::testing::matchers::impl::NodeMatcherProperties 131 void set_control_deps(NodeSeqMatcher control_deps) { 133 control_deps_ = std::move(control_deps); 143 !control_deps().has_value() && !attr().has_value(); 163 absl::Span<const ::testing::Matcher<const Node*>> control_deps); 219 // Matches a node with control dependences `control_deps`. 221 // `control_deps` are unordered and will match the control deps of a node in any 224 impl::NodeMatcherProperties CtrlDeps(Ts... control_deps) { 225 return impl::CtrlDeps({control_deps...});
|
increase_dynamism_for_auto_jit_pass.cc | 149 const std::vector<const Edge*>& control_deps) 150 : scope_(s), control_deps_(control_deps) {} 174 std::vector<const Edge*> control_deps, Output* size) { 194 ConstantCache constant_pool(host_scope, control_deps); 224 for (const Edge* e : control_deps) {
|
node_matchers.cc | 176 std::vector<const Node*> control_deps; variable 179 control_deps.push_back(e->src()); 185 !control_dep_set->MatchAndExplain(control_deps, &inner_listener)) { 416 if (prop.control_deps()) { 419 ::testing::UnorderedElementsAreArray(*prop.control_deps()); 462 absl::Span<const ::testing::Matcher<const Node*>> control_deps) { 464 absl::c_copy(control_deps, std::back_inserter(control_deps_vector));
|
encapsulate_util.cc | 275 std::vector<string> control_deps; local 278 &control_deps); 287 for (const string& control_input : control_deps) {
|
/external/tensorflow/tensorflow/cc/framework/ |
scope.h | 126 /// control dependencies the union of operations in the control_deps vector 129 const gtl::ArraySlice<Operation>& control_deps) const; 232 const std::vector<Operation>& control_deps() const;
|
scope_internal.h | 48 const std::vector<Operation>& control_deps() const { return control_deps_; } function in class:tensorflow::Scope::Impl 74 std::vector<Operation> control_deps, bool clear_control_deps);
|
scope_test.cc | 143 EXPECT_EQ(c.control_deps().size(), 2); 145 EXPECT_EQ(c_c.control_deps().size(), 3);
|
scope.cc | 120 std::vector<Operation> control_deps, bool clear_control_deps) 129 : (control_deps.insert(control_deps.begin(), 132 control_deps)), 298 const std::vector<Operation>& Scope::control_deps() const { function in class:tensorflow::Scope 430 const gtl::ArraySlice<Operation>& control_deps) const { 433 std::vector<Operation>(control_deps.begin(), control_deps.end()),
|
/external/tensorflow/tensorflow/core/common_runtime/ |
constant_folding.cc | 290 gtl::FlatSet<Node*>& control_deps = (*constant_control_deps)[n]; local 301 control_deps.insert(e->src()); 308 control_deps.insert(parent_deps.begin(), parent_deps.end()); 465 // 'control_deps' is the set of nodes that should be control predecessors of the 469 const Tensor& constant, const gtl::FlatSet<Node*>& control_deps, 548 if (control_deps.empty()) { 551 for (Node* node : control_deps) { 646 const gtl::FlatSet<Node*>& control_deps = local 650 control_deps, opts.max_constant_size_in_bytes, generate_new_name)) {
|
/external/tensorflow/tensorflow/contrib/compiler/ |
xla.py | 347 output_tensors, control_deps = _postprocess_flat_outputs(outputs) 349 output_tensors, control_deps = _postprocess_non_flat_outputs(outputs) 361 return control_flow_ops.group(control_deps, name='output_0') 368 with ops.control_dependencies(control_deps):
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
function_optimizer.cc | 257 absl::flat_hash_set<string> control_deps; member in struct:tensorflow::grappler::__anon45146::FunctionSpecialization 554 absl::flat_hash_set<string>* control_deps) { 555 // Record node control dependencies in the control_deps set. 560 control_deps->insert(input); 603 if (!specialization.control_deps.empty()) { 610 for (const string& ctrl : specialization.control_deps) { 815 absl::flat_hash_set<string> control_deps local [all...] |
/external/tensorflow/tensorflow/python/tpu/ |
tpu.py | [all...] |
tpu_estimator.py | [all...] |
/external/tensorflow/tensorflow/c/ |
c_api.cc | [all...] |