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

Allow use of REGISTER_UNARY_*_FUNCTION* macros outside namespace tensorflow

Browse Source 
PiperOrigin-RevId: 254221618
This commit is contained in:
A. Unique TensorFlower 2019-06-20 10:18:36 -07:00 committed by TensorFlower Gardener
parent ffd4de40fb
commit 063d42680b
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/framework/variant_op_registry.h
View File

@ -438,8 +438,8 @@ class UnaryVariantBinaryOpRegistration {
REGISTER_UNARY_VARIANT_DECODE_FUNCTION_UNIQ(ctr, T, type_name)
#define REGISTER_UNARY_VARIANT_DECODE_FUNCTION_UNIQ(ctr, T, type_name) \
static variant_op_registry_fn_registration::UnaryVariantDecodeRegistration< \
T> \
static ::tensorflow::variant_op_registry_fn_registration:: \
UnaryVariantDecodeRegistration<T> \
register_unary_variant_op_decoder_fn_##ctr(type_name)
// ****** NOTE ******
@ -508,8 +508,8 @@ class UnaryVariantBinaryOpRegistration {
#define REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION_UNIQ( \
ctr, op, device, T, type_index, unary_op_function) \
static variant_op_registry_fn_registration::UnaryVariantUnaryOpRegistration< \
T> \
static ::tensorflow::variant_op_registry_fn_registration:: \
UnaryVariantUnaryOpRegistration<T> \
register_unary_variant_op_decoder_fn_##ctr(op, device, type_index, \
unary_op_function)
@ -529,7 +529,7 @@ class UnaryVariantBinaryOpRegistration {
#define REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION_UNIQ( \
ctr, op, device, T, type_index, binary_op_function) \
static variant_op_registry_fn_registration:: \
static ::tensorflow::variant_op_registry_fn_registration:: \
UnaryVariantBinaryOpRegistration<T> \
register_unary_variant_op_decoder_fn_##ctr(op, device, type_index, \
binary_op_function)