Searched refs:AllowVariantTypes (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 48 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
52 .AllowVariantTypes()
57 REGISTER_XLA_OP(Name("StopGradient").AllowVariantTypes(), IdentityOp);
|
| D | function_ops.cc | 75 .AllowVariantTypes()
80 .AllowVariantTypes()
|
| 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 | aggregate_ops.cc | 94 REGISTER_XLA_OP(Name("AddN").AllowVariantTypes(), AddNOp);
|
| D | tensor_list_ops.cc | 279 .AllowVariantTypes(),
656 REGISTER_XLA_OP(Name("TensorListPushBack").AllowVariantTypes(),
692 REGISTER_XLA_OP(Name("TensorListPopBack").AllowVariantTypes(),
|
| 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(),
|
| D | shape_op.cc | 398 REGISTER_XLA_OP(Name("ZerosLike").AllowVariantTypes(), ZerosLikeOp);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 351 XlaOpRegistrationBuilder& AllowVariantTypes();
|
| D | xla_op_registry.cc | 518 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowVariantTypes() { in AllowVariantTypes() function in tensorflow::XlaOpRegistrationBuilder
|