Searched refs:AllowResourceTypes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 49 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
52 .AllowResourceTypes()
|
| D | function_ops.cc | 74 .AllowResourceTypes()
79 .AllowResourceTypes()
|
| D | retval_op.cc | 69 Name("_Retval").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | if_op.cc | 382 REGISTER_XLA_OP(Name("If").AllowResourceTypes().AllowVariantTypes(), XlaIfOp);
383 REGISTER_XLA_OP(Name("StatelessIf").AllowResourceTypes().AllowVariantTypes(),
385 REGISTER_XLA_OP(Name("XlaIf").AllowResourceTypes().AllowVariantTypes(),
|
| D | arg_op.cc | 116 Name("_Arg").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | case_op.cc | 364 REGISTER_XLA_OP(Name("Case").AllowResourceTypes().AllowVariantTypes(),
366 REGISTER_XLA_OP(Name("StatelessCase").AllowResourceTypes().AllowVariantTypes(),
|
| D | while_op.cc | 645 REGISTER_XLA_OP(Name("While").AllowResourceTypes().AllowVariantTypes(),
647 REGISTER_XLA_OP(Name("StatelessWhile").AllowResourceTypes().AllowVariantTypes(),
649 REGISTER_XLA_OP(Name("XlaWhile").AllowResourceTypes().AllowVariantTypes(),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 355 XlaOpRegistrationBuilder& AllowResourceTypes();
|
| D | xla_op_registry.cc | 513 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowResourceTypes() { in AllowResourceTypes() function in tensorflow::XlaOpRegistrationBuilder
|