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

Comparison with literal shape should ignore tiles.

Browse Source 
PiperOrigin-RevId: 243144930
This commit is contained in:
A. Unique TensorFlower 2019-04-11 14:39:51 -07:00 committed by TensorFlower Gardener
parent c80ba8dba5
commit 4e28bb323d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/service/hlo_evaluator.cc
View File

@ -423,7 +423,8 @@ Status HloEvaluator::HandleParameter(HloInstruction* parameter) {
#ifndef NDEBUG
const Literal* input_literal = arg_literals_[parameter->parameter_number()];
VLOG(2) << "Parameter evaluated to: " << input_literal->ToString();
DCHECK(ShapeUtil::Equal(parameter->shape(), input_literal->shape()))
DCHECK(Shape::Equal().MinorToMajorOnlyInLayout()(parameter->shape(),
input_literal->shape()))
<< "parameter shape is: "
<< ShapeUtil::HumanStringWithLayout(parameter->shape())
<< ", but input literal shape is: "