Internal Copybara change.

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
tensorflow
core
lite/delegates/gpu
gl/kernels
metal/kernels

View File

@ -28,7 +28,9 @@ limitations under the License.
// the configuration through the environment variable.
//
// 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)
@ -37,7 +39,11 @@ namespace internal {
// 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.
// 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() {
static bool use_custom_contraction_kernel = true;

View File

@ -33,7 +33,7 @@ limitations under the License.
// to your function, your callers will automatically convert bools, integers,
// 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
// AlphaNum &". In particular, instantiating AlphaNum directly as a stack
// variable is not supported.

View File

@ -93,7 +93,7 @@ class Resize : public NodeShader {
st.xy = max(icoord_floor, ivec2(0, 0));
st.zw = min(icoord_floor + ivec2(1, 1), borders);
vec2 t = coord - coord_floor; //interpolating factors
vec2 t = coord - coord_floor; // interpolating factors
vec4 tex11 = $input_data_0[st.x, st.y, gid.z]$;
vec4 tex21 = $input_data_0[st.z, st.y, gid.z]$;

View File

@ -54,7 +54,7 @@ std::string GetResizeBilinearCode(bool half_pixel_centers) {
int4 st;
st.xy = max(itex_coord_floor, int2(0, 0));
st.zw = min(itex_coord_floor + int2(1, 1), borders);
const float2 t = tex_coord - tex_coord_floor; //interpolating factors
const float2 t = tex_coord - tex_coord_floor; // interpolating factors
const int src_index0 = (gid.z * size.y + st.y) * size.x + st.x;
const int src_index1 = (gid.z * size.y + st.y) * size.x + st.z;
const int src_index2 = (gid.z * size.y + st.w) * size.x + st.x;