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

Mark the ExpandDimOp and the SqueezeOp as non expensive

Browse Source 
Change: 128892723
This commit is contained in:
Benoit Steiner 2016-07-30 07:23:05 -08:00 committed by TensorFlower Gardener
parent ac281d1df4
commit 6fa593f46d
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/shape_ops.cc
View File

@ -253,6 +253,8 @@ class ExpandDimsOp : public OpKernel {
" and output shape ", output_shape.DebugString()));
}
}
bool IsExpensive() override { return false; }
};
REGISTER_KERNEL_BUILDER(Name("ExpandDims").Device(DEVICE_CPU).HostMemory("dim"),
ExpandDimsOp);
@ -342,6 +344,8 @@ class SqueezeOp : public OpKernel {
}
}
bool IsExpensive() override { return false; }
private:
std::unordered_set<int32> squeeze_dims_;
};