Searched refs:AllowVariantTypes (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | identity_op.cc | 49 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
53 .AllowVariantTypes()
58 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 | 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 | aggregate_ops.cc | 94 REGISTER_XLA_OP(Name("AddN").AllowVariantTypes(), AddNOp);
|
| D | tensor_list_ops.cc | 285 .AllowVariantTypes(),
664 REGISTER_XLA_OP(Name("TensorListPushBack").AllowVariantTypes(),
700 REGISTER_XLA_OP(Name("TensorListPopBack").AllowVariantTypes(),
|
| 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(),
|
| D | shape_op.cc | 432 REGISTER_XLA_OP(Name("ZerosLike").AllowVariantTypes(), ZerosLikeOp);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_embedding_ops.cc | 116 REGISTER_XLA_OP(Name("XlaRecvTPUEmbeddingActivations").AllowVariantTypes(),
188 Name("XlaRecvTPUEmbeddingDeduplicationData").AllowVariantTypes(),
294 REGISTER_XLA_OP(Name("XlaSendTPUEmbeddingGradients").AllowVariantTypes(),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 358 XlaOpRegistrationBuilder& AllowVariantTypes();
|
| D | xla_op_registry.cc | 518 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::AllowVariantTypes() { in AllowVariantTypes() function in tensorflow::XlaOpRegistrationBuilder
|