Home
last modified time | relevance | path

Searched refs:IsMetadataOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc69 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);
Dtensor_list_ops.cc105 REGISTER_XLA_OP(Name("TensorListLength").IsMetadataOp(), TensorListLengthOp);
336 REGISTER_XLA_OP(Name("TensorListElementShape").IsMetadataOp(),
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h212 static bool IsMetadataOp(const string& op);
362 XlaOpRegistrationBuilder& IsMetadataOp();
Dxla_op_registry.cc446 /*static*/ bool XlaOpRegistry::IsMetadataOp(const string& op) { in IsMetadataOp() function in tensorflow::XlaOpRegistry
552 XlaOpRegistrationBuilder& XlaOpRegistrationBuilder::IsMetadataOp() { in IsMetadataOp() function in tensorflow::XlaOpRegistrationBuilder
Dconst_analysis.cc212 if (XlaOpRegistry::IsMetadataOp(node->type_string())) { in BackwardsConstAnalysis()