Searched refs:AllowResourceTypes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 48 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
51 .AllowResourceTypes()
|
| D | function_ops.cc | 74 .AllowResourceTypes()
79 .AllowResourceTypes()
|
| D | retval_op.cc | 69 Name("_Retval").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | if_op.cc | 357 REGISTER_XLA_OP(Name("If").AllowResourceTypes().AllowVariantTypes(), XlaIfOp);
358 REGISTER_XLA_OP(Name("StatelessIf").AllowResourceTypes().AllowVariantTypes(),
360 REGISTER_XLA_OP(Name("XlaIf").AllowResourceTypes().AllowVariantTypes(),
|
| D | arg_op.cc | 89 Name("_Arg").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | case_op.cc | 372 REGISTER_XLA_OP(Name("Case").AllowResourceTypes().AllowVariantTypes(),
374 REGISTER_XLA_OP(Name("StatelessCase").AllowResourceTypes().AllowVariantTypes(),
|
| D | while_op.cc | 631 REGISTER_XLA_OP(Name("While").AllowResourceTypes().AllowVariantTypes(),
633 REGISTER_XLA_OP(Name("StatelessWhile").AllowResourceTypes().AllowVariantTypes(),
635 REGISTER_XLA_OP(Name("XlaWhile").AllowResourceTypes().AllowVariantTypes(),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 348 XlaOpRegistrationBuilder& AllowResourceTypes();
|
| D | xla_op_registry.cc | 513 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowResourceTypes() { in AllowResourceTypes() function in tensorflow::XlaOpRegistrationBuilder
|