| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | math_grad_test.cc | 106 GradientRegistry registry_; member in tensorflow::gradients::internal::__anon4b0d0d6e0111::CppGradients
133 status_ = registry_.Register("AddV2", AddRegisterer); in TEST_P()
137 AddModel, BuildGradModel(AddModel, registry_), in TEST_P()
151 status_ = registry_.Register("Exp", ExpRegisterer); in TEST_P()
155 ExpModel, BuildGradModel(ExpModel, registry_), in TEST_P()
187 status_ = registry_.Register("MatMul", MatMulRegisterer); in TEST_P()
201 MatMulModel, BuildGradModel(MatMulModel, registry_), in TEST_P()
217 status_ = registry_.Register("Sqrt", SqrtRegisterer); in TEST_P()
221 SqrtModel, BuildGradModel(SqrtModel, registry_), in TEST_P()
235 status_ = registry_.Register("Neg", NegRegisterer); in TEST_P()
[all …]
|
| /external/libchrome/mojo/public/cpp/bindings/lib/ |
| D | sync_handle_watcher.cc | 20 registry_(SyncHandleRegistry::current()), in SyncHandleWatcher()
26 registry_->UnregisterHandle(handle_); in ~SyncHandleWatcher()
48 bool result = registry_->Wait(should_stop_array, 2); in SyncWatch()
62 registry_->RegisterHandle(handle_, handle_signals_, callback_); in IncrementRegisterCount()
71 registry_->UnregisterHandle(handle_); in DecrementRegisterCount()
|
| D | sync_event_watcher.cc | 18 registry_(SyncHandleRegistry::current()), in SyncEventWatcher()
24 registry_->UnregisterEvent(event_, callback_); in ~SyncEventWatcher()
51 bool result = registry_->Wait(should_stop_array.container().data(), in SyncWatch()
65 registry_->RegisterEvent(event_, callback_); in IncrementRegisterCount()
74 registry_->UnregisterEvent(event_, callback_); in DecrementRegisterCount()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | logger_registry.cc | 30 if (!registry_.emplace(name, std::unique_ptr<nvinfer1::ILogger>(logger)) in Register()
39 const auto found = registry_.find(name); in LookUp()
40 if (found == registry_.end()) { in LookUp()
49 registry_ TF_GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/core/lib/monitoring/ |
| D | collection_registry.cc | 75 const auto found_it = registry_.find(metric_def->name()); in Register()
76 if (found_it != registry_.end()) { in Register()
81 registry_.insert( in Register()
91 registry_.erase(metric_def->name()); in Unregister()
99 for (const auto& registration : registry_) { in CollectMetrics()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op.cc | 56 for (const auto& e : registry_) delete e.second; in ~OpRegistry()
95 gtl::FindWithDefault(registry_, op_type_name, nullptr)) { in LookUp()
112 res = gtl::FindWithDefault(registry_, op_type_name, nullptr); in LookUpSlow()
142 for (const auto& p : registry_) { in GetRegisteredOps()
151 for (const auto& p : registry_) { in GetOpRegistrationData()
171 registry_.begin(), registry_.end()); in Export()
240 !gtl::InsertIfNotPresent(®istry_, op_reg_data->op_def.name(), in RegisterAlreadyLocked()
|
| D | op.h | 163 mutable std::unordered_map<string, const OpRegistrationData*> registry_
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 134 per_pass_state.reserve(registry_->passes().size()); in Run()
138 for (const auto& pass_registration : registry_->passes()) { in Run()
177 << "(registered " << registry_->passes().size() << ")"; in Run()
186 << ", Total: " << registry_->passes().size(); in Run()
222 for (auto& pass_registration : registry_->passes()) { in Run()
332 absl::c_any_of(registry_->passes(), [&](auto& pass_registration) -> bool { in Run()
341 << "(registered " << registry_->passes().size() << " passes)"; in Run()
346 << "(registered " << registry_->passes().size() in Run()
365 for (auto& pass_registration : registry_->passes()) { in Run()
|
| D | mlir_graph_optimization_pass.h | 118 : registry_(registry) {} in registry_() function
140 const MlirOptimizationPassRegistry* registry_;
206 : registry_(registry) {} in registry_() function
211 const MlirV1CompatOptimizationPassRegistry* registry_;
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | grad_op_registry.cc | 28 CHECK(registry_.insert({op, func}).second) << "Existing gradient for " << op; in Register()
33 auto iter = registry_.find(op); in Lookup()
34 if (iter == registry_.end()) { in Lookup()
|
| D | gradients.cc | 103 const ops::GradOpRegistry* registry_; member in tensorflow::__anon810028dc0111::SymbolicGradientBuilder
146 registry_(registry), in SymbolicGradientBuilder()
376 Status s = registry_->Lookup(opname, &grad_fn); in IsPrimitiveOpWithNoGrad()
385 TF_RETURN_IF_ERROR(registry_->Lookup(op.node()->type_string(), &grad_fn)); in CallGradFunction()
|
| D | grad_op_registry.h | 54 std::unordered_map<string, GradFunc> registry_;
|
| /external/tensorflow/tensorflow/c/experimental/gradients/tape/ |
| D | tape_context.cc | 24 : AbstractContext(kTape), parent_ctx_(c), tape_(tape), registry_(registry) { in TapeContext()
37 return new TapeOperation(parent_ctx_->CreateOperation(), tape_, registry_); in CreateOperation()
|
| D | tape_context.h | 39 const GradientRegistry& registry_; variable
|
| D | tape_operation.cc | 27 registry_(registry) { in TapeOperation()
211 TF_RETURN_IF_ERROR(registry_.Lookup(forward_op_, &backward_fn)); in Execute()
|
| D | tape_operation.h | 75 const GradientRegistry& registry_; variable
|
| /external/libchrome/components/policy/core/common/ |
| D | mock_configuration_policy_provider.h | 40 ConfigurationPolicyProvider::Init(®istry_); in Init()
50 SchemaRegistry registry_; variable
|
| /external/libchrome/mojo/public/cpp/bindings/tests/ |
| D | versioning_test_service.cc | 100 registry_.AddInterface<HumanResourceDatabase>( in HumanResourceSystemServer()
108 registry_.BindInterface(interface_name, std::move(interface_pipe)); in OnBindInterface()
118 service_manager::BinderRegistry registry_; member in mojo::test::versioning::HumanResourceSystemServer
|
| D | sync_handle_registry_unittest.cc | 17 SyncHandleRegistryTest() : registry_(SyncHandleRegistry::current()) {} in SyncHandleRegistryTest()
19 const scoped_refptr<SyncHandleRegistry>& registry() { return registry_; } in registry()
22 scoped_refptr<SyncHandleRegistry> registry_; member in mojo::SyncHandleRegistryTest
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | env.cc | 67 mutable std::unordered_map<std::string, std::unique_ptr<FileSystem>> registry_
74 if (!registry_.emplace(scheme, std::unique_ptr<FileSystem>(factory())) in Register()
85 if (!registry_.emplace(scheme, std::move(filesystem)).second) { in Register()
94 const auto found = registry_.find(scheme); in Lookup()
95 if (found == registry_.end()) { in Lookup()
104 for (const auto& e : registry_) { in GetRegisteredFileSystemSchemes()
|
| /external/gflags/src/ |
| D | gflags.cc | 929 explicit CommandLineFlagParser(FlagRegistry* reg) : registry_(reg) {} in CommandLineFlagParser()
981 FlagRegistry* const registry_; member in GFLAGS_NAMESPACE::__anon4cd27d730111::CommandLineFlagParser
1034 registry_->Lock(); in ParseNewCommandLineFlags()
1059 CommandLineFlag* flag = registry_->SplitArgumentLocked(arg, &key, &value, in ParseNewCommandLineFlags()
1107 registry_->Unlock(); in ParseNewCommandLineFlags()
1148 CommandLineFlag* flag = registry_->FindFlagLocked(flagname); in ProcessFromenvLocked()
1184 if (value && !registry_->SetFlagLocked(flag, value, set_mode, &msg)) { in ProcessSingleOptionLocked()
1207 FlagRegistryLock frl(registry_); in ValidateFlags()
1208 for (FlagRegistry::FlagConstIterator i = registry_->flags_.begin(); in ValidateFlags()
1209 i != registry_->flags_.end(); ++i) { in ValidateFlags()
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients.cc | 56 auto iter = registry_.find(op_name); in Register()
57 if (iter != registry_.end()) { in Register()
61 registry_.insert({op_name, gradient_function_factory}); in Register()
67 auto iter = registry_.find(op.op_name); in Lookup()
68 if (iter == registry_.end()) { in Lookup()
|
| D | gradients.h | 86 absl::flat_hash_map<string, GradientFunctionFactory> registry_;
|
| /external/libchrome/mojo/public/cpp/bindings/ |
| D | sync_handle_watcher.h | 64 scoped_refptr<SyncHandleRegistry> registry_; variable
|
| D | sync_event_watcher.h | 62 scoped_refptr<SyncHandleRegistry> registry_; variable
|