Home
last modified time | relevance | path

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

/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 …]
Dcompactptrset.h30 using BigRep = FlatSet<T>;
/external/tensorflow/tensorflow/core/grappler/
Dop_types.cc97 static const gtl::FlatSet<string>* const kCastLikeOps = in IsCastLike()
98 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsCastLike()
185 static const gtl::FlatSet<string>* const kMonotonicNonDecreasingOps = in IsElementWiseMonotonic()
186 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsElementWiseMonotonic()
192 static const gtl::FlatSet<string>* const kMonotonicNonIncreasingOps = in IsElementWiseMonotonic()
193 CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Neg", "Rsqrt"})); in IsElementWiseMonotonic()
516 static const gtl::FlatSet<string>* const kTensorArrayOps = in IsTensorArray()
517 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsTensorArray()
697 static const gtl::FlatSet<string>* const kInvolutionOps = in OPDEF_PROPERTY_HELPER()
698 CHECK_NOTNULL((new gtl::FlatSet<string>{"Conj", "Reciprocal", "Invert", in OPDEF_PROPERTY_HELPER()
[all …]
Dutils.h99 gtl::FlatSet<T, Hash> set_;
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding.cc268 std::unordered_map<const Node*, gtl::FlatSet<Node*>>* constant_control_deps, in ConsiderConstantFoldableNode()
290 gtl::FlatSet<Node*>& control_deps = (*constant_control_deps)[n]; in ConsiderConstantFoldableNode()
306 const gtl::FlatSet<Node*>& parent_deps = in ConsiderConstantFoldableNode()
325 std::unordered_map<const Node*, gtl::FlatSet<Node*>>* constant_control_deps, in FindConstantFoldableNodes()
469 const Tensor& constant, const gtl::FlatSet<Node*>& control_deps, in ReplaceTensorWithConstant()
581 std::unordered_map<const Node*, gtl::FlatSet<Node*>> constant_control_deps; in ConstantFold()
646 const gtl::FlatSet<Node*>& control_deps = in ConstantFold()
Dgraph_execution_state.cc407 static const gtl::FlatSet<string>* const kHasExplicitShapeAttribute = in GetFeedShapeAndTypeFromAttribute()
408 CHECK_NOTNULL((new gtl::FlatSet<string>{ in GetFeedShapeAndTypeFromAttribute()
Dexecutor.cc321 gtl::FlatSet<string> unique_frame_names;
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer_test.cc32 gtl::FlatSet<string> devices = {}; in TEST_F()
38 gtl::FlatSet<string> devices = {"/device:CPU:0", "/device:XLA_GPU:0"}; in TEST_F()
48 gtl::FlatSet<string> devices = {"/device:XLA_CPU:0", "/device:XLA_GPU:0"}; in TEST_F()
58 gtl::FlatSet<string> devices = {"/device:XLA_GPU:0"}; in TEST_F()
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;
Dpin_to_host_optimizer.cc268 string TryFindHostDevice(const gtl::FlatSet<string>& devices, in TryFindHostDevice()
316 gtl::FlatSet<string> devices; in Optimize()
Darithmetic_optimizer.h135 gtl::FlatSet<string> feed_nodes_;
Dfunction_optimizer_test.cc1776 gtl::FlatSet<string> expected_ctrl = {"^init", "^xf"}; in TEST_F()
1777 gtl::FlatSet<string> actual_ctrl = {node.input(1), node.input(2)}; in TEST_F()
/external/tensorflow/tensorflow/c/eager/
Dtape.h326 const gtl::FlatSet<int64>& sources_set, bool persistent_tape) { in PrepareBackprop()
457 gtl::FlatMap<string, gtl::FlatSet<int>>* FunctionsAcceptingNoneForIndicesMap() { in FunctionsAcceptingNoneForIndicesMap()
458 static auto* const m = new gtl::FlatMap<string, gtl::FlatSet<int>>({ in FunctionsAcceptingNoneForIndicesMap()
482 gtl::FlatSet<int64> sources_set(source_tensor_ids.begin(), in ComputeGradient()
646 gtl::FlatSet<int64> used_gradient_ids(source_tensor_ids.size()); in ComputeGradient()
Dc_api_internal.h108 tensorflow::gtl::FlatSet<std::string> attrs; // attributes inferred so far
/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dsubgraph.h35 class Identity : public gtl::FlatSet<const GenNode*> {
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.h196 gtl::FlatSet<BaseRecvTensorCall*> active_ GUARDED_BY(mu_);
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1698 tensorflow::gtl::FlatSet<tensorflow::int64> sources_set(sources_vec.begin(),
1755 tensorflow::gtl::FlatSet<PyObject*> seen_results(result.size());
1916 std::pair<bool, tensorflow::gtl::FlatSet<int>>** output) {
1918 string, std::pair<bool, tensorflow::gtl::FlatSet<int>>>* m =
1920 string, std::pair<bool, tensorflow::gtl::FlatSet<int>>>({
2014 std::pair<bool, tensorflow::gtl::FlatSet<int>>** output) {
2016 string, std::pair<bool, tensorflow::gtl::FlatSet<int>>>* m =
2018 string, std::pair<bool, tensorflow::gtl::FlatSet<int>>>({
2057 PyObject* seq, const tensorflow::gtl::FlatSet<int>& indices) {
2094 std::pair<bool, tensorflow::gtl::FlatSet<int>>* outputs_not_required;
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc71 gtl::FlatSet<string> GetNodeNamesSet(const Iterable& nodes) { in GetNodeNamesSet()
74 gtl::FlatSet<string> names; in GetNodeNamesSet()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h328 gtl::FlatSet<uint64> active_remote_contexts_;
Dexecute.cc787 static const gtl::FlatSet<string>* unpinnable_ops = new gtl::FlatSet<string>({ in IsPinnableOp()
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor.cc276 gtl::FlatSet<StringPiece, StringPieceHasher> gdef_prefixes_;
282 gtl::FlatSet<StringPiece, StringPieceHasher> existing_prefixes_;
363 gtl::FlatSet<StringPiece, StringPieceHasher>* prefixes) { in AddPrefixes()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc464 static const gtl::FlatSet<DataType>* const kRealNumberTypes = in IsNumericType()
465 CHECK_NOTNULL((new gtl::FlatSet<DataType>{ in IsNumericType()
493 static const gtl::FlatSet<string>* const kOpTpeWhitelist = in IsWhiteListedOpTypeForEvaluateNode()
494 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsWhiteListedOpTypeForEvaluateNode()
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc508 gtl::FlatSet<int64> previous_indices; in ComputeSquaredNormPerExample()