experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
Katherine Tian 2020-08-12 17:44:55 +00:00
parent 61d63b3764
commit 9220304a54
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.bazelversion
View File

@ -1 +1 @@
3.0.0 3.1.0

9
tensorflow/core/kernels/map_kernels.h
View File

@ -152,8 +152,8 @@ class TensorMapErase : public OpKernel {
OP_REQUIRES(c, m->tensors().find(key) != m->tensors().end(), OP_REQUIRES(c, m->tensors().find(key) != m->tensors().end(),
errors::InvalidArgument("Trying to erase non-existent item.")); errors::InvalidArgument("Trying to erase non-existent item."));
const Tensor& t = m->tensors().find(key)->second; //const Tensor& t = m->tensors().find(key)->second;
c->set_output(1, t); //c->set_output(1, t);
TensorMap* output_map = nullptr; TensorMap* output_map = nullptr;
OP_REQUIRES_OK(c, ForwardInputOrCreateNewMap(c, 0, 0, *m, &output_map)); OP_REQUIRES_OK(c, ForwardInputOrCreateNewMap(c, 0, 0, *m, &output_map));
@ -179,7 +179,8 @@ class TensorMapHasKey : public OpKernel {
template <typename Device> template <typename Device>
Status TensorMapBinaryAdd(OpKernelContext* c, const TensorMap& a, Status TensorMapBinaryAdd(OpKernelContext* c, const TensorMap& a,
const TensorMap& b, TensorMap* out) { const TensorMap& b, TensorMap* out) {
// binary add returns a union of keys, values with keys in the intersection are added // Binary add returns a map containing the union of keys.
// Values with keys in the intersection are added.
out->tensors() = a.tensors(); out->tensors() = a.tensors();
for (const std::pair<TensorKey,Tensor>& p : b.tensors()) { for (const std::pair<TensorKey,Tensor>& p : b.tensors()) {
absl::flat_hash_map<TensorKey,Tensor>::iterator it = out->tensors().find(p.first); absl::flat_hash_map<TensorKey,Tensor>::iterator it = out->tensors().find(p.first);
@ -197,7 +198,7 @@ Status TensorMapBinaryAdd(OpKernelContext* c, const TensorMap& a,
template <typename Device> template <typename Device>
Status TensorMapZerosLike(OpKernelContext* c, const TensorMap& x, TensorMap* y) { Status TensorMapZerosLike(OpKernelContext* c, const TensorMap& x, TensorMap* y) {
// zeros like returns an empty map // Zeros like returns an empty map.
return Status::OK(); return Status::OK();
} }