Internal Copybara change.
PiperOrigin-RevId: 302071704 Change-Id: If250e336a66aa462b09881cf9a8f6eb4e6ccc3a7
This commit is contained in:
parent
a794c690b4
commit
289af050cc
tensorflow
core
lite/delegates/gpu
@ -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;
|
||||||
|
|
||||||
|
@ -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.
|
||||||
|
@ -93,7 +93,7 @@ class Resize : public NodeShader {
|
|||||||
st.xy = max(icoord_floor, ivec2(0, 0));
|
st.xy = max(icoord_floor, ivec2(0, 0));
|
||||||
st.zw = min(icoord_floor + ivec2(1, 1), borders);
|
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 tex11 = $input_data_0[st.x, st.y, gid.z]$;
|
||||||
vec4 tex21 = $input_data_0[st.z, st.y, gid.z]$;
|
vec4 tex21 = $input_data_0[st.z, st.y, gid.z]$;
|
||||||
|
@ -54,7 +54,7 @@ std::string GetResizeBilinearCode(bool half_pixel_centers) {
|
|||||||
int4 st;
|
int4 st;
|
||||||
st.xy = max(itex_coord_floor, int2(0, 0));
|
st.xy = max(itex_coord_floor, int2(0, 0));
|
||||||
st.zw = min(itex_coord_floor + int2(1, 1), borders);
|
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_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_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;
|
const int src_index2 = (gid.z * size.y + st.w) * size.x + st.x;
|
||||||
|
Loading…
Reference in New Issue
Block a user