Searched refs:AllowResourceTypes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 42 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
45 .AllowResourceTypes()
|
| D | function_ops.cc | 111 REGISTER_XLA_OP(Name("PartitionedCall").AllowResourceTypes(),
113 REGISTER_XLA_OP(Name("StatefulPartitionedCall").AllowResourceTypes(),
|
| D | if_op.cc | 283 REGISTER_XLA_OP(Name("If").AllowResourceTypes(), XlaIfOp);
284 REGISTER_XLA_OP(Name("StatelessIf").AllowResourceTypes(), XlaIfOp);
285 REGISTER_XLA_OP(Name("XlaIf").AllowResourceTypes(), XlaIfOp);
|
| D | retval_op.cc | 67 Name("_Retval").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | arg_op.cc | 71 Name("_Arg").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | while_op.cc | 578 REGISTER_XLA_OP(Name("While").AllowResourceTypes().AllowVariantTypes(),
580 REGISTER_XLA_OP(Name("StatelessWhile").AllowResourceTypes().AllowVariantTypes(),
582 REGISTER_XLA_OP(Name("XlaWhile").AllowResourceTypes().AllowVariantTypes(),
|
| D | case_op.cc | 295 REGISTER_XLA_OP(Name("Case").AllowResourceTypes(), XlaCaseOp);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 298 XlaOpRegistrationBuilder& AllowResourceTypes();
|
| D | xla_op_registry.cc | 492 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowResourceTypes() { in AllowResourceTypes() function in tensorflow::XlaOpRegistrationBuilder
|