Searched refs:register_op (Results 1 – 4 of 4) sorted by relevance
470 void TestFunction(const RegisterFunctionOp& register_op, in TestFunction() argument488 *enqueue_request.add_queue()->mutable_register_function() = register_op; in TestFunction()570 void TestComponentFunction(const RegisterFunctionOp& register_op, in TestComponentFunction() argument587 *enqueue_request.add_queue()->mutable_register_function() = register_op; in TestComponentFunction()665 RegisterFunctionOp register_op; in TEST_F() local666 *register_op.mutable_function_def() = MatMulFunction(); in TEST_F()667 TestFunction(register_op, "MatMulFunction"); in TEST_F()671 RegisterFunctionOp register_op; in TEST_F() local672 *register_op.mutable_function_def() = MatMulFunction(); in TEST_F()673 TestFunction(register_op, "MatMulFunction", /*local_inputs=*/true); in TEST_F()[all …]
215 namespace register_op {278 static ::tensorflow::InitOnStartupMarker const register_op##ctr \281 << ::tensorflow::register_op::OpDefBuilderWrapper(name)
292 namespace register_op { namespace
325 __u8 register_op; /* IORING_RESTRICTION_REGISTER_OP */ member