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

Internal Copybara change.

Browse Source 
PiperOrigin-RevId: 302071704
Change-Id: If250e336a66aa462b09881cf9a8f6eb4e6ccc3a7
This commit is contained in:
Cesar Crusius 2020-03-20 11:57:09 -07:00 committed by TensorFlower Gardener
parent a794c690b4
commit 289af050cc
4 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/kernels/eigen_contraction_kernel.cc
View File

@ -28,7 +28,9 @@ limitations under the License.
// the configuration through the environment variable. // the configuration through the environment variable.
// //
// Example: // Example:
// bazel test --test_env=TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL=false //test // bazel test \
// --test_env=TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL=false \
// //path/to:test
#if defined(TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL) #if defined(TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL)
@ -37,7 +39,11 @@ namespace internal {
// TODO(ezhulenev): This is a temporary workaround for disabling custom kernels // TODO(ezhulenev): This is a temporary workaround for disabling custom kernels
// at runtime in tests. We should always rely on compile time flags for that. // at runtime in tests. We should always rely on compile time flags for that.
// Example: ... --test_env=TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL=false //test //
// Example:
// bazel test \
// --test_env=TENSORFLOW_USE_CUSTOM_CONTRACTION_KERNEL=false \
// //path/to:test
EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE bool UseCustomContractionKernels() { EIGEN_DEVICE_FUNC EIGEN_DONT_INLINE bool UseCustomContractionKernels() {
static bool use_custom_contraction_kernel = true; static bool use_custom_contraction_kernel = true;

2
tensorflow/core/platform/strcat.h
View File

@ -33,7 +33,7 @@ limitations under the License.
// to your function, your callers will automatically convert bools, integers, // to your function, your callers will automatically convert bools, integers,
// and floating point values to strings for you. // and floating point values to strings for you.
// //
// NOTE: Use of AlphaNum outside of the //strings package is unsupported except // NOTE: Use of AlphaNum outside of the "strings" package is unsupported except
// for the specific case of function parameters of type "AlphaNum" or "const // for the specific case of function parameters of type "AlphaNum" or "const
// AlphaNum &". In particular, instantiating AlphaNum directly as a stack // AlphaNum &". In particular, instantiating AlphaNum directly as a stack
// variable is not supported. // variable is not supported.