Home
last modified time | relevance | path

Searched refs:FlatSet (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision_lists.h36 virtual gtl::FlatSet<string> AllowList() = 0;
40 virtual gtl::FlatSet<string> InferList() = 0;
45 virtual gtl::FlatSet<string> DenyList() = 0;
49 virtual gtl::FlatSet<string> ClearList() = 0;
53 static void UpdateList(const string& list_name, gtl::FlatSet<string>* list) { in UpdateList()
75 static void AddTensorListOps(gtl::FlatSet<string>* list) { in AddTensorListOps()
110 gtl::FlatSet<string> AllowList() override { in AllowList()
111 auto list = gtl::FlatSet<string>{ in AllowList()
158 gtl::FlatSet<string> InferList() override { in InferList()
160 return gtl::FlatSet<string>{}; in InferList()
[all …]
Dpin_to_host_optimizer.h29 string TryFindHostDevice(const gtl::FlatSet<string>& devices,
Dgraph_optimizer_stage.h52 gtl::FlatSet<string>* feed_nodes, in GraphOptimizerContext()
65 gtl::FlatSet<string>* feed_nodes;
Dauto_mixed_precision.cc845 Status ValidateLists(const gtl::FlatSet<string>& allow_list, in ValidateLists()
846 const gtl::FlatSet<string>& deny_list, in ValidateLists()
847 const gtl::FlatSet<string>& infer_list, in ValidateLists()
848 const gtl::FlatSet<string>& clear_list) { in ValidateLists()
849 std::vector<gtl::FlatSet<string>> lists{allow_list, deny_list, infer_list, in ValidateLists()
1010 gtl::FlatSet<string> f16_allowlist_;
1011 gtl::FlatSet<string> f16_denylist_;
1012 gtl::FlatSet<string> f16_inferlist_;
1013 gtl::FlatSet<string> f16_clearlist_;
1180 static const gtl::FlatSet<string> tensor_list_reader_ops = { in IsTensorListReaderOp()
[all …]
Dpin_to_host_optimizer.cc273 string TryFindHostDevice(const gtl::FlatSet<string>& devices, in TryFindHostDevice()
311 gtl::FlatSet<string> devices; in Optimize()
/external/tensorflow/tensorflow/core/lib/gtl/
Dflatset.h38 class FlatSet {
55 FlatSet() : FlatSet(1) {} in FlatSet() function
57 explicit FlatSet(size_t N, const Hash& hf = Hash(), const Eq& eq = Eq())
60 FlatSet(const FlatSet& src) : rep_(src.rep_) {} in FlatSet() function
64 FlatSet(FlatSet&& src) : rep_(std::move(src.rep_)) {} in FlatSet() function
67 FlatSet(InputIter first, InputIter last, size_t N = 1,
69 : FlatSet(N, hf, eq) { in FlatSet() function
73 FlatSet(std::initializer_list<value_type> init, size_t N = 1,
75 : FlatSet(init.begin(), init.end(), N, hf, eq) {}
77 FlatSet& operator=(const FlatSet& src) {
[all …]
Dflatset_test.cc29 typedef FlatSet<int64> NumSet;
226 TEST(FlatSet, ClearNoResize) { in TEST() argument
236 TEST(FlatSet, Clear) { in TEST() argument
246 TEST(FlatSet, Copy) { in TEST() argument
260 TEST(FlatSet, InitFromIter) { in TEST() argument
270 TEST(FlatSet, InitializerList) { in TEST() argument
282 TEST(FlatSet, InsertIter) { in TEST() argument
293 TEST(FlatSet, Eq) { in TEST() argument
316 TEST(FlatSet, Swap) { in TEST() argument
329 TEST(FlatSet, Reserve) { in TEST() argument
[all …]
/external/libchrome/base/containers/
Dflat_set_unittest.cc24 TEST(FlatSet, IncompleteType) { in TEST() argument
38 TEST(FlatSet, RangeConstructor) { in TEST() argument
46 TEST(FlatSet, MoveConstructor) { in TEST() argument
59 TEST(FlatSet, InitializerListConstructor) { in TEST() argument
64 TEST(FlatSet, InsertFindSize) { in TEST() argument
76 TEST(FlatSet, CopySwap) { in TEST() argument
94 TEST(FlatSet, UsingTransparentCompare) { in TEST() argument
/external/tensorflow/tensorflow/python/eager/
Dpywrap_gradient_exclusions.cc40 auto *m = new tensorflow::gtl::FlatMap<string, tensorflow::gtl::FlatSet<int>>; in OpGradientInfoInit()
43 tensorflow::gtl::FlatSet<int>( in OpGradientInfoInit()
51 absl::optional<tensorflow::gtl::FlatSet<int>> OpGradientUnusedInputIndices( in OpGradientUnusedInputIndices()
422 absl::optional<tensorflow::gtl::FlatSet<int>> OpGradientUnusedOutputIndices( in OpGradientUnusedOutputIndices()
Dpywrap_gradient_exclusions.h27 absl::optional<tensorflow::gtl::FlatSet<int>> OpGradientUnusedInputIndices(
35 absl::optional<tensorflow::gtl::FlatSet<int>> OpGradientUnusedOutputIndices(
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc127 static const gtl::FlatSet<string>* const kCastLikeOps = in IsCastLike()
128 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsCastLike()
227 static const gtl::FlatSet<string>* const kMonotonicNonDecreasingOps = in IsElementWiseMonotonic()
228 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsElementWiseMonotonic()
234 static const gtl::FlatSet<string>* const kMonotonicNonIncreasingOps = in IsElementWiseMonotonic()
235 CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Neg", "Rsqrt"})); in IsElementWiseMonotonic()
597 static const gtl::FlatSet<string>* const kTensorArrayOps = in IsTensorArray()
598 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsTensorArray()
788 static const gtl::FlatSet<string>* const kInvolutionOps = in OPDEF_PROPERTY_HELPER()
789 CHECK_NOTNULL((new gtl::FlatSet<string>{"Conj", "Reciprocal", "Invert", in OPDEF_PROPERTY_HELPER()
[all …]
/external/perfetto/src/traced/probes/ftrace/
Dftrace_metadata.h145 base::FlatSet<InodeBlockPair> inode_and_device;
146 base::FlatSet<int32_t> rename_pids;
147 base::FlatSet<int32_t> pids;
148 base::FlatSet<KernelAddr> kernel_addrs;
Dprintk_formats_parser.h63 base::FlatSet<PrintkEntry> set_;
/external/perfetto/src/profiling/perf/
Devent_config.h47 base::FlatSet<std::string> cmdlines;
48 base::FlatSet<std::string> exclude_cmdlines;
49 base::FlatSet<pid_t> pids;
50 base::FlatSet<pid_t> exclude_pids;
Dunwinding.cc156 base::FlatSet<DataSourceInstanceID> pending_sample_sources = in ProcessQueue()
161 base::FlatSet<DataSourceInstanceID> sources_to_stop; in ProcessQueue()
193 base::FlatSet<DataSourceInstanceID> Unwinder::ConsumeAndUnwindReadySamples() { in ConsumeAndUnwindReadySamples()
195 base::FlatSet<DataSourceInstanceID> pending_sample_sources; in ConsumeAndUnwindReadySamples()
/external/perfetto/include/perfetto/base/
Dflat_set.h42 class FlatSet {
49 FlatSet() = default;
52 FlatSet(std::initializer_list<T> initial) : entries_(initial) { in FlatSet() function
/external/perfetto/src/traced/probes/ps/
Dprocess_stats_data_source.h63 void OnPids(const base::FlatSet<int32_t>& pids);
64 void OnRenamePids(const base::FlatSet<int32_t>& pids);
118 void WriteProcessTree(const base::FlatSet<int32_t>&);
150 base::FlatSet<int32_t> seen_pids_;
Dprocess_stats_data_source.cc177 void ProcessStatsDataSource::OnPids(const base::FlatSet<int32_t>& pids) { in OnPids()
184 const base::FlatSet<int32_t>& pids) { in WriteProcessTree()
198 void ProcessStatsDataSource::OnRenamePids(const base::FlatSet<int32_t>& pids) { in OnRenamePids()
399 base::FlatSet<int32_t> pids; in WriteAllProcessStats()
/external/tensorflow/tensorflow/core/graph/
Dedgeset.h52 auto s = new gtl::FlatSet<const Edge*>(new_size); in reserve()
72 gtl::FlatSet<const Edge*>* get_set() const { in get_set()
74 return static_cast<gtl::FlatSet<const Edge*>*>( in get_set()
115 typename gtl::FlatSet<const Edge*>::const_iterator tree_iter_;
Dedgeset.cc40 s = new gtl::FlatSet<const Edge*>; in insert()
/external/tensorflow/tensorflow/core/common_runtime/
Dinput_colocation_exemption_registry.h43 const gtl::FlatSet<string>& Get() { return ops_; } in Get()
49 gtl::FlatSet<string> ops_;
Dconstant_folding.cc294 std::unordered_map<const Node*, gtl::FlatSet<Node*>>* constant_control_deps, in ConsiderConstantFoldableNode()
317 gtl::FlatSet<Node*>& control_deps = (*constant_control_deps)[n]; in ConsiderConstantFoldableNode()
333 const gtl::FlatSet<Node*>& parent_deps = in ConsiderConstantFoldableNode()
352 std::unordered_map<const Node*, gtl::FlatSet<Node*>>* constant_control_deps, in FindConstantFoldableNodes()
496 const Tensor& constant, const gtl::FlatSet<Node*>& control_deps, in ReplaceTensorWithConstant()
609 std::unordered_map<const Node*, gtl::FlatSet<Node*>> constant_control_deps; in ConstantFold()
675 const gtl::FlatSet<Node*>& control_deps = in ConstantFold()
/external/perfetto/src/base/
Dflat_set_unittest.cc32 FlatSet<long> flat_set; in TEST()
87 flat_set = FlatSet<long>({1, 2, 1, 2, 3}); in TEST()
93 FlatSet<int> flat_set; in TEST()
Dflat_set_benchmark.cc61 using perfetto::base::FlatSet;
62 BENCHMARK_TEMPLATE(BM_SetInsert, FlatSet<int>)->Apply(BenchmarkArgs);
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_utils.cc34 static const gtl::FlatSet<string>* unpinnable_ops = new gtl::FlatSet<string>({ in IsPinnableOp()

12