Searched refs:TensorKey (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_map.h | 106 absl::flat_hash_map<TensorKey, Tensor>& tensors() { in tensors()
110 const absl::flat_hash_map<TensorKey, Tensor>& tensors() const { in tensors()
124 bool insert(const TensorKey& key, const Tensor& value) { in insert()
130 absl::flat_hash_map<TensorKey, Tensor>::iterator find(TensorKey key) { in find()
134 Tensor& lookup(TensorKey key) { return tensors_->values_.find(key)->second; } in lookup()
136 Tensor& operator[](TensorKey& k) { return tensors_->values_[k]; }
138 bool replace(const TensorKey& k, const Tensor& v) { in replace()
144 size_t erase(TensorKey key) { return tensors_->values_.erase(key); } in erase()
152 absl::flat_hash_map<TensorKey, Tensor>::iterator it = in keys()
168 absl::flat_hash_map<TensorKey, Tensor> values_;
|
| D | tensor_map_test.cc | 35 TensorKey k1 = Tensor(15); in TEST()
36 TensorKey k2 = Tensor(15); in TEST()
41 TensorKey k3 = Tensor(37.0); in TEST()
48 TensorKey k = Tensor(11); in TEST()
51 absl::flat_hash_map<TensorKey, Tensor> am; in TEST()
54 absl::flat_hash_map<TensorKey, Tensor>::iterator map_it = in TEST()
64 TensorKey k = Tensor(11); in TEST()
67 absl::flat_hash_map<TensorKey, Tensor>::iterator map_it = tm.find(k); in TEST()
76 TensorKey k = Tensor(11); in TEST()
85 TensorKey k = Tensor(11); in TEST()
[all …]
|
| D | tensor_map.cc | 32 absl::flat_hash_map<TensorKey, Tensor>::const_iterator map_it = in Encode()
49 for (const std::pair<TensorKey, Tensor>& p : from.tensors()) { in TensorMapDeviceCopy()
50 TensorKey to_key(p.first.dtype()); in TensorMapDeviceCopy()
|
| D | map_kernels.h | 113 const TensorKey& key = ctx->input(1); in Compute()
133 const TensorKey& key = ctx->input(1); in Compute()
150 const TensorKey& key = ctx->input(1); in Compute()
173 const TensorKey& key = ctx->input(1); in Compute()
229 for (const std::pair<TensorKey, Tensor>& p : b.tensors()) { in TensorMapBinaryAdd()
230 absl::flat_hash_map<TensorKey, Tensor>::iterator it = in TensorMapBinaryAdd()
|
| D | ragged_tensor_variant.h | 97 if (TensorKey(x.splits(i)) != TensorKey(y.splits(i))) { in RaggedTensorVariantBinaryAdd()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_key.h | 22 class TensorKey : public Tensor {
26 TensorKey(const Tensor& t) : Tensor(t) {} in TensorKey() function
29 friend bool operator==(const TensorKey& t1, const TensorKey& t2) {
49 friend bool operator!=(const TensorKey& t1, const TensorKey& t2) {
55 friend H AbslHashValue(H h, const TensorKey& k) { in AbslHashValue()
|