Skip to content

Commit

Permalink
Temporarily fix bias_op_gpu.h std::tuple problem
Browse files Browse the repository at this point in the history
  • Loading branch information
aaroey committed Jun 22, 2018
1 parent 937489b commit 93896fb
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions tensorflow/core/kernels/bias_op_gpu.h
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,12 @@ class BiasAddParams {
hash_code_ = Hash64Combine(hash_code_, device_id);
}
bool operator==(const BiasAddParams& other) const {
return this->get_data_as_tuple() == other.get_data_as_tuple();
// return this->get_data_as_tuple() == other.get_data_as_tuple();
return
this->in_shape_ == other.in_shape_ &&
this->data_format_ == other.data_format_ &&
this->dtype_ == other.dtype_ &&
this->device_id_ == other.device_id_;
}

bool operator!=(const BiasAddParams& other) const {
Expand All @@ -114,9 +119,9 @@ class BiasAddParams {
protected:
using ParamsDataType = std::tuple<SpatialArray, TensorFormat, DataType, int>;

ParamsDataType get_data_as_tuple() const {
return std::make_tuple(in_shape_, data_format_, dtype_, device_id_);
}
// ParamsDataType get_data_as_tuple() const {
// return std::make_tuple(in_shape_, data_format_, dtype_, device_id_);
// }

uint64 hash_code_ = 0;

Expand Down

0 comments on commit 93896fb

Please sign in to comment.