| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | flatset.h | 38 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 …]
|
| D | flatset_test.cc | 29 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 …]
|
| D | compactptrset.h | 30 using BigRep = FlatSet<T>;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.cc | 97 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 …]
|
| D | utils.h | 99 gtl::FlatSet<T, Hash> set_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding.cc | 268 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()
|
| D | graph_execution_state.cc | 407 static const gtl::FlatSet<string>* const kHasExplicitShapeAttribute = in GetFeedShapeAndTypeFromAttribute()
408 CHECK_NOTNULL((new gtl::FlatSet<string>{ in GetFeedShapeAndTypeFromAttribute()
|
| D | executor.cc | 321 gtl::FlatSet<string> unique_frame_names;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer_test.cc | 32 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()
|
| D | pin_to_host_optimizer.h | 29 string TryFindHostDevice(const gtl::FlatSet<string>& devices,
|
| D | graph_optimizer_stage.h | 52 gtl::FlatSet<string>* feed_nodes, in GraphOptimizerContext()
65 gtl::FlatSet<string>* feed_nodes;
|
| D | pin_to_host_optimizer.cc | 268 string TryFindHostDevice(const gtl::FlatSet<string>& devices, in TryFindHostDevice()
316 gtl::FlatSet<string> devices; in Optimize()
|
| D | arithmetic_optimizer.h | 135 gtl::FlatSet<string> feed_nodes_;
|
| D | function_optimizer_test.cc | 1776 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/ |
| D | tape.h | 326 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()
|
| D | c_api_internal.h | 108 tensorflow::gtl::FlatSet<std::string> attrs; // attributes inferred so far
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | subgraph.h | 35 class Identity : public gtl::FlatSet<const GenNode*> {
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.h | 196 gtl::FlatSet<BaseRecvTensorCall*> active_ GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 1698 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/ |
| D | fusion_utils.cc | 71 gtl::FlatSet<string> GetNodeNamesSet(const Iterable& nodes) { in GetNodeNamesSet()
74 gtl::FlatSet<string> names; in GetNodeNamesSet()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 328 gtl::FlatSet<uint64> active_remote_contexts_;
|
| D | execute.cc | 787 static const gtl::FlatSet<string>* unpinnable_ops = new gtl::FlatSet<string>({ in IsPinnableOp()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_constructor.cc | 276 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/ |
| D | graph_properties.cc | 464 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/ |
| D | sdca_internal.cc | 508 gtl::FlatSet<int64> previous_indices; in ComputeSquaredNormPerExample()
|