Searched refs:CompilationOnly (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | no_op.cc | 34 REGISTER_XLA_OP(Name("NoOp").CompilationOnly(), NoOp);
38 REGISTER_XLA_OP(Name("ControlTrigger").CompilationOnly(), NoOp);
|
| D | identity_op.cc | 48 Name("Identity").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
53 .CompilationOnly(),
|
| D | stack_ops.cc | 124 Name("StackV2").CompileTimeConstantInput("max_size").CompilationOnly(),
172 REGISTER_XLA_OP(Name("StackPushV2").CompilationOnly(), StackPushOp);
226 REGISTER_XLA_OP(Name("StackPopV2").CompilationOnly(), StackPopOp);
240 REGISTER_XLA_OP(Name("StackCloseV2").CompilationOnly(), StackCloseOp);
|
| D | variable_ops.cc | 62 REGISTER_XLA_OP(Name("VariableShape").CompilationOnly(), VariableShapeOp);
80 REGISTER_XLA_OP(Name("ReadVariableOp").CompilationOnly(), ReadVariableOp);
90 REGISTER_XLA_OP(Name("AssignVariableOp").CompilationOnly(), AssignVariableOp);
|
| D | shape_op.cc | 69 REGISTER_XLA_OP(Name("Shape").CompilationOnly().IsMetadataOp(), ShapeOp);
183 REGISTER_XLA_OP(Name("ShapeN").CompilationOnly().IsMetadataOp(), ShapeNOp);
199 REGISTER_XLA_OP(Name("Rank").CompilationOnly().IsMetadataOp(), RankOp);
226 REGISTER_XLA_OP(Name("Size").CompilationOnly().IsMetadataOp(), SizeOp);
|
| D | assert_op.cc | 46 REGISTER_XLA_OP(Name("Assert").CompilationOnly(), AssertOp);
|
| D | retval_op.cc | 69 Name("_Retval").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | arg_op.cc | 89 Name("_Arg").AllowResourceTypes().AllowVariantTypes().CompilationOnly(),
|
| D | const_op.cc | 133 REGISTER_XLA_OP(Name("Const").CompilationOnly(), ConstOp);
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | aot_only_var_handle_op.cc | 81 REGISTER_XLA_OP(Name(tfcompile::kXlaAotOnlyVarHandleOp).CompilationOnly(),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.h | 345 XlaOpRegistrationBuilder& CompilationOnly();
|
| D | xla_op_registry.cc | 508 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::CompilationOnly() { in CompilationOnly() function in tensorflow::XlaOpRegistrationBuilder
|